I wrote a Notebook [0] that gets you introduced to JAX in a very gentle manner. It also covers things like functional purity in Deep Learning, and handling of

I wrote a Notebook [0] that gets you introduced to JAX in a very gentle manner. It also covers things like functional purity in Deep Learning, and handling of random numbers in JAX. [0]: Learn JAX: From Linear Regression to Neural Networks - https://www.kaggle.com/code/truthr/jax-0 https://www.kaggle.com/code/truthr/jax-0