Getting Started
jax.debug.print
jax.debug.breakpoint
checkify
Reference Documentation
Advanced JAX Tutorials
Notes
Developer documentation
API documentation
Hard \(\mathrm{tanh}\) activation function.
Computes the element-wise function:
x (Any) – input array
Any