jax.default\_backend ==================== .. currentmodule:: jax .. autofunction:: default_backend