JAX for Machine Learning: how it works and why learn it

An introduction to JAX, its best features alongside with code snippets for you to get started

Learning about JAX :axes in vmap()

JAX Crash Course - Accelerating Machine Learning code!

Ivy: The Unified Machine Learning Framework - Decibel

Learning Basic Electronics

Tutorial 2 (JAX): Introduction to JAX+Flax — UvA DL Notebooks v1.2

Lecture 2: Development Infrastructure & Tooling - The Full Stack

Open-Source Machine Learning in Computational Chemistry

Tensorflow Extended (TFX) in action: build a production ready deep learning pipeline

Useful Algorithms That Are Not Optimized By Jax, PyTorch, or