This page cannot be shown here. You can still read it on the original site — the toolbar below keeps your place in the directory.
Recently I’ve been trying to learn more about JAX, the next-gen ML framework
from DeepMind. These are the resources I’ve found most helpful so far. 
 
 The JAX docs , which are excellent 
 Patrick Kidger’s Learning JAX as a PyTorch developer 
 JAX AI Stack tutorials 
 The book JAX in Action , along with accompanying Python notebooks 
 Codebases: Gemma , MaxText…
Comments
Nothing yet. Say the first thing.
Sign in to join the conversation.