6 ms·Hi, JAX: An introduction to vanilla JAX for deep learning research2 points by matomatical 10mo ago