jax-foo Trying out the JAX library on some examples. Notebook trying a ResNet model on CIFAR 10 using JAX. Based off this repo.