Strong agree with @ricardoV94 – if you’re getting started with PyMC, you’ll find the best support, most responsive issue response, fastest bug fixes, and a huge amount of existing utility functions by just using the nuts_sampler="numpyro". Once you call bx.Model.from_pymc(model), your work downstream from there will be using the JAX ecosystem.
It suffices to be aware that you can usually add one line from bayeux and have access to some different samplers (and optimizers and VI).