JAX Course - 2. Working with Neural Networks in JAX

📍 Q&A and forum discussion: In this meetup, Sanyam will walk us through Neural Networks in JAX and we’ll understand a paper implementation done in JAX.