Getting Started
Part 1: Introduction
Part 2: Custom Tasks, Task Families, and Performance Improvements
Part 3: Truncation and TruncatedStep
Part 4: GradientEstimators
Part 5: Meta-training with GradientLearner
Part 6: Custom learned optimizer architectures
Tutorials
No dependency introduction to learned optimizers in JAX
Summary tutorial: Getting metrics out of your models
Modules
Tasks
Optimizers
Learned Optimizers
Outer Trainers
Continuous Eval
Population: Online hparam adjustment for multiple workers
Optimizer Baselines
learned_optimization
Python Module Index
Python Module Index
l
l
learned_optimization
learned_optimization.optimizers.base