A library within the Jax AI stack focused on optimizations, particularly for creating optimizers.
DeepLearningAI