Comment by wjessup

Great basics tutorial: https://colinraffel.com/blog/you-don-t-know-jax.html