Jax Jax Resources EI Seminar: Matthew Johnson: JAX: accelerated ML research via composable function transformations Other Machine Learning