Speed Up Your Code by 50x: A Guide to Moving from NumPy to JAX

I created and taught (alongside my brother!) participants about various features from Jax that allowed me to speed up my code by over 100x. All the materials are open and can be found at numpy-to-jax