Issues
- 2
Recommended neos setup
#20 opened by alexander-held - 0
fix transforms
#19 opened by lukasheinrich - 2
Facilitate gradient interoperability between other ml frameworks (TensorFlow, Torch, etc) and jax
#3 opened by phinate - 4
- 3
Update to new upcoming `fax` API
#16 opened by gehring - 3
celluloid doesn't install with requirements
#12 opened by andrzejnovak - 2
demo_training not fully executable
#11 opened by andrzejnovak - 0
Refactor parts of demos/demo_training.ipynb into the library, and make good tests
#5 opened by phinate - 0
Develop a realistic use case
#6 opened by phinate - 0
Speed up training by making use of `jax.jit`
#2 opened by phinate