Repository navigation
FlatPPL alpha 8
Pre-releaseAlpha wheels for JAX/jaxlib 0.11.2 and Python 3.12–3.14 on macOS arm64, Linux x86_64 and Linux aarch64.
This release includes saved-state reuse in reverse scans, upstream compiler updates through 5f463e91, and the upstream GPU SVD correction. Four separate fixes preserve reduction semantics, distinct reduction axes, unsigned update indices through reshape, and inactive masks through differentiable arithmetic.
Validation: 1,571 native tests passed, with nine expected failures. All nine wheel ABIs pass. Values, gradients, JVPs, nested scans and Hessian-vector products pass on M4 CPU, A100, GH200 and GB10 in float32 and float64. The FlatPPL package passes 284 CPU tests and 267 A100 tests, with one optional GPU skip. Sixteen Polars-dependent tests ran on CPU only. Documentation doctests and all six gallery examples pass.
Warm batched float64 variance value-and-gradient time on M4 falls from 3.80–3.91 µs with alpha 7 to 1.34 µs, versus native JAX at 1.42 µs. GPU gains vary by workload. These bounded checks do not establish a general speedup.
SHA256SUMS and provenance.json identify the exact wheel files and tested source tree.