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
  • Index

Index

A | L | M | O | S

A

  • Adam() (in module learned_optimization.optimizers.base)

L

  • learned_optimization.optimizers.base
    • module

M

  • module
    • learned_optimization.optimizers.base

O

  • Optimizer() (in module learned_optimization.optimizers.base)

S

  • SGD() (in module learned_optimization.optimizers.base)
  • SGDM() (in module learned_optimization.optimizers.base)

© Copyright 2021, Google LLC..

Built with Sphinx using a theme provided by Read the Docs.