Tutorials
Advanced JAX Tutorials
Notes
Developer documentation
API documentation
jit
vmap
pmap
jax.numpy.
int32
__init__
Initialize self. See help(type(self)) for accurate signature.
Methods
Initialize self.
Attributes
dtype