Introduction to JAX (AI Adventures)

JAX is an open-source Python library that brings together Autograd and XLA, facilitating high-performance machine learning research. In this episode of AI Adventures, Yufeng Guo talks about how you can use JAX to compile and run your NumPy programs on GPUs and TPUs, and its ability to support composable function transformations.

GitHub for Jax →
Try out the notebook used in this video →
XLA: Optimizer for machine learning →

Check out the rest of the Cloud AI Adventures playlist →
Subscribe to get all the episodes as they come out →

Product: AI Platform Training; fullname: Yufeng Guo;

Duration: 00:07:05
Publisher: Google Cloud
You can watch this video also at the source.

Inxy Hosting CDN Marketplace