Jax

Jax Resources

• EI Seminar: Matthew Johnson: JAX: accelerated ML research via composable function transformations
• The JAX Book: Scaling Machine Learning Systems With JAX

Other

• Machine Learning