Skip to content
UPEEC

JAX

NumPy-like arrays with automatic differentiation, JIT compilation and GPU/TPU execution.

Advanced
Comments 0

jax.grad(f) gives derivatives automatically; widely used in ML research and physics simulation.

Spotted a mistake or want to add detail? Suggest an improvement — Content Contributors' edits are reviewed by editors before they go live.

Discussion (0)

Sign in or join free to ask questions and discuss this topic.