Jax

Build and Train an LLM with JAX

DeepLearningAI

Numerical computing library enabling gradients and multi-device training