Replace deprecated jax context manager and JIT-traced non-array arguments#260
Open
hannes-holey wants to merge 3 commits intodfm:mainfrom
Open
Replace deprecated jax context manager and JIT-traced non-array arguments#260hannes-holey wants to merge 3 commits intodfm:mainfrom
hannes-holey wants to merge 3 commits intodfm:mainfrom
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}