Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Warning
This page was created from a pull request (#9655).
Elementwise arc tangent of two variables: \(\mathrm{atan}({x \over y})\).
x (Any) –
Any
y (Any) –
previous
jax.lax.atan
next
jax.lax.batch_matmul