ml

Contents

  • About
  • Getting Started

Configs

  • Launchers
  • LR Schedulers
  • Models
  • Optimizers

Reference

  • Core
  • Logger
  • LR Scheduler
  • Model
  • Optimizer
    • ml.optimizers.adam
    • ml.optimizers.adan
    • ml.optimizers.base
    • ml.optimizers.common
    • ml.optimizers.gan
    • ml.optimizers.lion
    • ml.optimizers.sgd
    • ml.optimizers.shampoo
    • ml.optimizers.types
  • Task
  • Trainer
  • Launcher
  • Utility Functions
ml
  • Optimizer
  • View page source

Optimizer

Contents:

  • ml.optimizers.adam
    • AdamOptimizerConfig
    • AdamOptimizer
  • ml.optimizers.adan
    • Adan
    • AdanOptimizerConfig
    • AdanOptimizer
  • ml.optimizers.base
    • BaseOptimizerConfig
    • BaseOptimizer
  • ml.optimizers.common
    • separate_decayable_params()
    • can_use_fused()
    • can_use_foreach()
  • ml.optimizers.gan
    • GenerativeAdversarialNetworkOptimizerConfig
    • GenerativeAdversarialNetworkOptimizer
  • ml.optimizers.lion
    • get_update_fn()
    • Lion
    • LionOptimizerConfig
    • LionOptimizer
  • ml.optimizers.sgd
    • SGDOptimizerConfig
    • SGDOptimizer
  • ml.optimizers.shampoo
    • Shampoo
    • ShampooOptimizerConfig
    • ShampooOptimizer
  • ml.optimizers.types
Previous Next

© Copyright 2023, Benjamin Bolte.

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