Optax

Software / App

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

Mentioned in 1 video

Videos Mentioning Optax

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

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

DeepLearningAI

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