site-packages/optax-0.2.8.dist-info/INSTALLER
site-packages/optax-0.2.8.dist-info/METADATA
site-packages/optax-0.2.8.dist-info/RECORD
site-packages/optax-0.2.8.dist-info/REQUESTED
site-packages/optax-0.2.8.dist-info/WHEEL
site-packages/optax-0.2.8.dist-info/direct_url.json
site-packages/optax-0.2.8.dist-info/licenses/LICENSE
site-packages/optax/CHANGELOG.md
site-packages/optax/__init__.py
site-packages/optax/_src/alias.py
site-packages/optax/_src/alias_test.py
site-packages/optax/_src/base.py
site-packages/optax/_src/base_test.py
site-packages/optax/_src/combine.py
site-packages/optax/_src/combine_test.py
site-packages/optax/_src/constrain.py
site-packages/optax/_src/deprecations.py
site-packages/optax/_src/factorized.py
site-packages/optax/_src/factorized_test.py
site-packages/optax/_src/float64_test.py
site-packages/optax/_src/linear_algebra.py
site-packages/optax/_src/linear_algebra_test.py
site-packages/optax/_src/linesearch.py
site-packages/optax/_src/linesearch_test.py
site-packages/optax/_src/lookahead.py
site-packages/optax/_src/lookahead_test.py
site-packages/optax/_src/numerics.py
site-packages/optax/_src/numerics_test.py
site-packages/optax/_src/schedule.py
site-packages/optax/_src/sharding_test.py
site-packages/optax/_src/test_utils.py
site-packages/optax/_src/transform.py
site-packages/optax/_src/transform_test.py
site-packages/optax/_src/update.py
site-packages/optax/_src/update_test.py
site-packages/optax/_src/utils.py
site-packages/optax/_src/utils_test.py
site-packages/optax/_src/wrappers.py
site-packages/optax/assignment/__init__.py
site-packages/optax/assignment/_hungarian_algorithm.py
site-packages/optax/assignment/_hungarian_algorithm_test.py
site-packages/optax/contrib/__init__.py
site-packages/optax/contrib/_acprop.py
site-packages/optax/contrib/_ademamix.py
site-packages/optax/contrib/_adopt.py
site-packages/optax/contrib/_cocob.py
site-packages/optax/contrib/_common_test.py
site-packages/optax/contrib/_complex_valued.py
site-packages/optax/contrib/_complex_valued_test.py
site-packages/optax/contrib/_dadapt_adamw.py
site-packages/optax/contrib/_dog.py
site-packages/optax/contrib/_galore.py
site-packages/optax/contrib/_galore_test.py
site-packages/optax/contrib/_madgrad.py
site-packages/optax/contrib/_mechanic.py
site-packages/optax/contrib/_mechanic_test.py
site-packages/optax/contrib/_momo.py
site-packages/optax/contrib/_muon.py
site-packages/optax/contrib/_muon_test.py
site-packages/optax/contrib/_privacy.py
site-packages/optax/contrib/_privacy_test.py
site-packages/optax/contrib/_prodigy.py
site-packages/optax/contrib/_reduce_on_plateau.py
site-packages/optax/contrib/_reduce_on_plateau_test.py
site-packages/optax/contrib/_sam.py
site-packages/optax/contrib/_sam_test.py
site-packages/optax/contrib/_schedule_free.py
site-packages/optax/contrib/_schedule_free_test.py
site-packages/optax/contrib/_sophia.py
site-packages/optax/experimental/__init__.py
site-packages/optax/experimental/_aggregating.py
site-packages/optax/experimental/_aggregating_test.py
site-packages/optax/losses/__init__.py
site-packages/optax/losses/_classification.py
site-packages/optax/losses/_classification_test.py
site-packages/optax/losses/_fenchel_young.py
site-packages/optax/losses/_fenchel_young_test.py
site-packages/optax/losses/_ranking.py
site-packages/optax/losses/_ranking_test.py
site-packages/optax/losses/_regression.py
site-packages/optax/losses/_regression_test.py
site-packages/optax/losses/_segmentation.py
site-packages/optax/losses/_segmentation_test.py
site-packages/optax/losses/_self_supervised.py
site-packages/optax/losses/_self_supervised_test.py
site-packages/optax/losses/_smoothing.py
site-packages/optax/losses/_smoothing_test.py
site-packages/optax/microbatching/__init__.py
site-packages/optax/microbatching/_microbatching.py
site-packages/optax/microbatching/_microbatching_test.py
site-packages/optax/optax_test.py
site-packages/optax/perturbations/__init__.py
site-packages/optax/perturbations/_make_pert.py
site-packages/optax/perturbations/_make_pert_test.py
site-packages/optax/projections/__init__.py
site-packages/optax/projections/_projections.py
site-packages/optax/projections/_projections_test.py
site-packages/optax/schedules/__init__.py
site-packages/optax/schedules/_inject.py
site-packages/optax/schedules/_inject_test.py
site-packages/optax/schedules/_join.py
site-packages/optax/schedules/_join_test.py
site-packages/optax/schedules/_schedule.py
site-packages/optax/schedules/_schedule_test.py
site-packages/optax/second_order/__init__.py
site-packages/optax/second_order/_deprecated.py
site-packages/optax/transforms/__init__.py
site-packages/optax/transforms/_accumulation.py
site-packages/optax/transforms/_accumulation_test.py
site-packages/optax/transforms/_adding.py
site-packages/optax/transforms/_adding_test.py
site-packages/optax/transforms/_clipping.py
site-packages/optax/transforms/_clipping_test.py
site-packages/optax/transforms/_combining.py
site-packages/optax/transforms/_combining_test.py
site-packages/optax/transforms/_conditionality.py
site-packages/optax/transforms/_conditionality_test.py
site-packages/optax/transforms/_constraining.py
site-packages/optax/transforms/_constraining_test.py
site-packages/optax/transforms/_freezing.py
site-packages/optax/transforms/_freezing_test.py
site-packages/optax/transforms/_layouts.py
site-packages/optax/transforms/_layouts_test.py
site-packages/optax/transforms/_masking.py
site-packages/optax/transforms/_masking_test.py
site-packages/optax/transforms/_monitoring.py
site-packages/optax/transforms/_monitoring_test.py
site-packages/optax/tree/__init__.py
site-packages/optax/tree_utils/__init__.py
site-packages/optax/tree_utils/_casting.py
site-packages/optax/tree_utils/_casting_test.py
site-packages/optax/tree_utils/_random.py
site-packages/optax/tree_utils/_random_test.py
site-packages/optax/tree_utils/_state_utils.py
site-packages/optax/tree_utils/_state_utils_test.py
site-packages/optax/tree_utils/_tree_math.py
site-packages/optax/tree_utils/_tree_math_test.py
