jax-experimental
jax-experimental copied to clipboard
Add "Run on Gradient" to notebooks?
It would be great to be able to run e.g. the MNIST example notebook on Paperspace with a "Run on gradient" button.