Hi, JAX: An introduction to vanilla JAX for deep learning researchgithub.com/matomatical2 pointsmatomatical5 months ago