Google Colab Update: PyMC3 is being replaced by PyMC v4 FAQ

Cool! Is there any additional step is we want to use Jax samplers with gpu? Thanks :blush: