Glossary · Math & training
JAX
A Python library for transforming numerical functions with automatic differentiation, compilation, vectorization, and parallel execution across accelerators. Its transformations work best with explicit state and functional-style code.
Common confusion
JAX does not prohibit all stateful programming, but hidden mutation inside transformed functions can produce incorrect or unsupported behavior.
Related terms
Sources
Browse the learning paths to see this term in context — every lesson is free to read.