JAX- Accelerated Machine Learning Research via Composable Function Transformations in Python

Explore accelerated machine learning research through composable function transformations in Python with JAX in this 52-minute seminar presented by Matthew Johnson from Google. Delivered as part of the Machine Learning Advances and Applications Seminar series at the Fields Institute, learn how JAX can enhance and streamline your machine learning workflows, offering powerful tools for researchers and practitioners alike.