AWhat changed
Combines Pyro-style effect handlers with JAX transformations and replaces NUTS's recursive tree-building implementation by an iterative form suitable for end-to-end JIT compilation.
Scientific scope
Combines Pyro-style effect handlers with JAX transformations and replaces NUTS's recursive tree-building implementation by an iterative form suitable for end-to-end JIT compilation. NUTS and JAX are inherited. Speedups vary by model, precision and hardware; time per leapfrog step is not automatically time per effective posterior sample, and finite sampling is not exact inference. First two authors contribute equally.
Evidence
The linked original sources describe the stated contribution and its evidence limits. WRS has not independently rerun the experiments or certified a proof. NUTS and JAX are inherited. Speedups vary by model, precision and hardware; time per leapfrog step is not automatically time per effective posterior sample, and finite sampling is not exact inference. First two authors contribute equally.
Contributors
- Source-listed authors and technical contributors · SCIENCE: Source-listed authors or explicitly credited technical contributors for the cited versions. Family entries retain source-specific attribution; a combined list is not one original-paper byline or equal individual work. Names alone do not establish resolved identities.
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