Hi, JAX: An introduction to vanilla JAX for deep learning research | Heykuki News