Refine JIT wrappers for new JAX for comaptiblity with `jax>=0.8.2`#809

Merged

chaoming0625 merged 3 commits intomasterfrom

Dec 25, 2025

Commits on Dec 25, 2025

refactor: clean up imports and remove unnecessary newlines
chaoming0625
committed
refactor: remove unused imports and clean up main execution block
chaoming0625
committed
feat: add brainpy_state module and update dependencies
chaoming0625
committed