Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Warning
This page was created from a pull request (#9655).
Context manager for jax_default_prng_impl config option.
Select the default PRNG implementation, used when one is not explicitly provided at seeding time.
previous
jax.default_matmul_precision
next
jax.numpy_rank_promotion