MENADD - Intro to JAX/Flax

Name: MENADD - Intro to JAX/Flax
Channel: Taha Bouhsine
Description: An accessible introduction to JAX and Flax for high-performance machine learning. Covers JAX's NumPy-style API, automatic differentiation, just-in-time (JIT) compilation, and vectorization, plus building and training neural networks with the Flax library.

@GDGMENA MENADD

About this Talk

An accessible introduction to JAX and Flax for high-performance machine learning. Covers JAX's NumPy-style API, automatic differentiation, just-in-time (JIT) compilation, and vectorization, plus building and training neural networks with the Flax library.

Resources

Slides