| Name | Modified | Size | Downloads / Week |
|---|---|---|---|
| Parent folder | |||
| 0.12.10 source code.tar.gz | 2026-09-24 | 15.7 MB | |
| 0.12.10 source code.zip | 2026-09-24 | 16.1 MB | |
| README.md | 2026-09-24 | 820 Bytes | |
| Totals: 3 Items | 31.8 MB | 0 | |
What's Changed
- verify Flax RoPE using mathematical definition instead of PyTorch by @copybara-service[bot] in https://github.com/google/flax/pull/5558
- Migrate all usages of HiPrimitive to VJPHiPrimitive and delete HiPrimitive by @copybara-service[bot] in https://github.com/google/flax/pull/5566
- [No functional change] Delete VJPHiPrimitive and replace with HiPrim. This is a simple replace. by @copybara-service[bot] in https://github.com/google/flax/pull/5568
- suppress pyrefly errors related to JAX scalar types by @copybara-service[bot] in https://github.com/google/flax/pull/5586
- Release Flax 0.12.10 and use attribute lookup for HiPrim fallback by @copybara-service[bot] in https://github.com/google/flax/pull/5597
Full Changelog: https://github.com/google/flax/compare/v0.12.9...v0.12.10
