Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Methods
__init__(self, arg0, **kwargs)
__init__
is_enabled()
is_enabled
set_metadata(self, **kwargs)
set_metadata