fix(jax): fix compatibility with flax 0.12#5067
Merged
wanghan-iapcm merged 11 commits intodeepmodeling:develfrom Dec 1, 2025
Merged
fix(jax): fix compatibility with flax 0.12#5067wanghan-iapcm merged 11 commits intodeepmodeling:develfrom
wanghan-iapcm merged 11 commits intodeepmodeling:develfrom
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}