JAX Tutorials#
This website presents introductory tutorials on JAX, designed and written by Smit Lunagariya.
JAX is a powerful library that revolutionizes numerical computing with its innovative approach to automatic differentiation and efficient execution on CPU, GPU, and TPU. By seamlessly combining ease of use with high performance, JAX empowers researchers and developers to unlock new frontiers in machine learning and scientific computing.
Lectures
References#
This lecture series uses some of the examples and inspiration from the following sources