JAX: accelerated machine learning research via composable function transformations in Python

Matthew Johnson, Google February 14, 2022 Machine Learning Advances and Applications Seminar (http://www.fields.utoronto.ca/activities/21-22/machine-learning)