AI Agent Skills

A hyperparameter optimization framework

Optax is a gradient processing and optimization library for JAX.

Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.