API Reference

The library is organized into three modules: core (types, state, base class), algorithm implementations, and utility functions for pytree operations and hypergradient computation.

• Core (core)
  • Types
  • Optimization State
  • Base Optimizer
• Algorithms (algorithms)
  • OnlineHypergradientOptimizer
  • T1T2Optimizer
  • GreedyOptimizer
  • Baselines
• Utils (utils)
  • tree_l2_norm()
  • tree_normalize()
  • tree_dot()
  • tree_zeros_like()
  • tree_lerp()
  • vjp_wrt_lambda()
  • vjp_wrt_both()
  • update_w_star()