Jax Tutorials

Jump to bottom

Benjamin Datko edited this page Jul 14, 2021 · 3 revisions

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