WRSWorld Research StateContribute
WORLD RESEARCH STATE

JAX: high-level tracing of pure array subroutines

Contribution grade

Differentiable-programming compiler systems · Human-designed research method or resource

Reference score205,000Provisional editorial estimate of this exact contribution; its grade is assessed separately.

What changed

Traced annotated pure-and-statically-composed Python/NumPy subroutines into XLA computations while preserving dynamic Python orchestration and compatibility with Autograd transformations.

Scientific scope

Traced annotated pure-and-statically-composed Python/NumPy subroutines into XLA computations while preserving dynamic Python orchestration and compatibility with Autograd transformations. OriginalSysML2018three-person report, not current project-maintainer citation. Inherits Autograd/NumPy/XLA; not invention of automatic differentiation or every later JAX transformation. Shape/type changes can recompile; staged subroutines obey purity/static-composition assumptions. Warmed-up speedups exclude potentially slow compilation; no arbitrary Python acceleration guarantee.

Evidence

The linked original sources describe the stated contribution and its evidence limits. WRS has not independently rerun the experiments or certified a proof. OriginalSysML2018three-person report, not current project-maintainer citation. Inherits Autograd/NumPy/XLA; not invention of automatic differentiation or every later JAX transformation. Shape/type changes can recompile; staged subroutines obey purity/static-composition assumptions. Warmed-up speedups exclude potentially slow compilation; no arbitrary Python acceleration guarantee.

Contributors

Provisional WRS assessment, edition 2026-10-16.1. Independent expert calibration is pending. This grade does not certify correctness or assign ownership.

Full scientific record · Browse contribution grades · Request a correction