Jax

Build and Train an LLM with JAX

DeepLearningAI

Numerical computing library enabling gradients and multi-device training

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

DeepLearningAI

Information Theory for Language Models: Jack Morris

Latent Space

A framework that some people switch from to learn CUDA.