Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
jit
vmap
pmap
jax.lax.
ge
Elementwise greater-than-or-equals: \(x \geq y\).
x (Any) –
Any
y (Any) –