In this course, instructor Janani Ravi gives you an in-depth look at JAX, a new experimental Python library designed for high performance, scientific computing and machine learning. Janani takes you through all aspects of JAX and what it is capable of, including: just-in-time compilation; automatic vectorization and automatic parallelization; computing gradients; performing transformations on pytrees; training simple neural networks; and more.
Learn More
