Jax Tutorials - bdatko/numpyro_play GitHub Wiki

JAX, aka NumPy on steroids

Getting started with JAX (MLPs, CNNs & RNNs)

Deterministic Randomness

• The site accompanies a workshop available on YouTube titled TUTORIAL / Eric Ma / Magical NumPy with JAX

Introduction to JAX for Machine Learning and More