Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
jit
vmap
pmap
jax.numpy.
cdouble
alias of jax._src.numpy.lax_numpy.complex128
jax._src.numpy.lax_numpy.complex128