Introduction to JAX for Machine Learning and More
This workshop will be an Introduction to JAX for Machine Learning and More, hosted by our very own DSC Exec, Nicholas Vadivelu.
JAX is a numeric computing library with a numpy-like interface. It specializes in function transformations such as auto-differentiation, just in time compilation (via XLA), and more. Through this workshop, you'll learn about what JAX is, how to use it, how it is different from other libraries like PyTorch and TensorFlow, why you should consider adding it to your toolkit.