Tutorials
Advanced JAX Tutorials
Notes
Developer documentation
API documentation
jit
vmap
pmap
Warning
This page was created from a pull request.
jax.numpy.
double
alias of jax._src.numpy.lax_numpy.float64
jax._src.numpy.lax_numpy.float64