lib/python3.9/site-packages/docs/__pycache__/conf.cpython-39.pyc
lib/python3.9/site-packages/docs/conf.py
lib/python3.9/site-packages/docs/ext/__pycache__/coverage_check.cpython-39.pyc
lib/python3.9/site-packages/docs/ext/coverage_check.py
lib/python3.9/site-packages/examples/__pycache__/datasets.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/datasets_test.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/differentially_private_sgd.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/flax_example.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/haiku_example.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/lookahead_mnist.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/lookahead_mnist_test.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/mnist.cpython-39.pyc
lib/python3.9/site-packages/examples/__pycache__/mnist_test.cpython-39.pyc
lib/python3.9/site-packages/examples/datasets.py
lib/python3.9/site-packages/examples/datasets_test.py
lib/python3.9/site-packages/examples/differentially_private_sgd.py
lib/python3.9/site-packages/examples/flax_example.py
lib/python3.9/site-packages/examples/haiku_example.py
lib/python3.9/site-packages/examples/lookahead_mnist.py
lib/python3.9/site-packages/examples/lookahead_mnist_test.py
lib/python3.9/site-packages/examples/mnist.py
lib/python3.9/site-packages/examples/mnist_test.py
lib/python3.9/site-packages/optax-0.1.4.dist-info/INSTALLER
lib/python3.9/site-packages/optax-0.1.4.dist-info/LICENSE
lib/python3.9/site-packages/optax-0.1.4.dist-info/METADATA
lib/python3.9/site-packages/optax-0.1.4.dist-info/RECORD
lib/python3.9/site-packages/optax-0.1.4.dist-info/REQUESTED
lib/python3.9/site-packages/optax-0.1.4.dist-info/WHEEL
lib/python3.9/site-packages/optax-0.1.4.dist-info/direct_url.json
lib/python3.9/site-packages/optax-0.1.4.dist-info/top_level.txt
lib/python3.9/site-packages/optax/__init__.py
lib/python3.9/site-packages/optax/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/optax/__pycache__/optax_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/alias.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/alias_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/base.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/base_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/clipping.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/clipping_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/combine.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/combine_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/constrain.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/constrain_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/control_variates.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/control_variates_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/equivalence_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/factorized.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/factorized_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/float64_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/linear_algebra.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/linear_algebra_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/lookahead.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/lookahead_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/loss.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/loss_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/numerics.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/numerics_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/privacy.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/privacy_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/schedule.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/schedule_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/second_order.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/second_order_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/stochastic_gradient_estimators.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/stochastic_gradient_estimators_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/test_utils.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/transform.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/transform_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/update.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/update_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/utils.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/utils_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/wrappers.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/__pycache__/wrappers_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/alias.py
lib/python3.9/site-packages/optax/_src/alias_test.py
lib/python3.9/site-packages/optax/_src/base.py
lib/python3.9/site-packages/optax/_src/base_test.py
lib/python3.9/site-packages/optax/_src/clipping.py
lib/python3.9/site-packages/optax/_src/clipping_test.py
lib/python3.9/site-packages/optax/_src/combine.py
lib/python3.9/site-packages/optax/_src/combine_test.py
lib/python3.9/site-packages/optax/_src/constrain.py
lib/python3.9/site-packages/optax/_src/constrain_test.py
lib/python3.9/site-packages/optax/_src/control_variates.py
lib/python3.9/site-packages/optax/_src/control_variates_test.py
lib/python3.9/site-packages/optax/_src/equivalence_test.py
lib/python3.9/site-packages/optax/_src/experimental/__pycache__/complex_valued.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/experimental/__pycache__/complex_valued_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/experimental/__pycache__/extra_args.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/experimental/__pycache__/extra_args_test.cpython-39.pyc
lib/python3.9/site-packages/optax/_src/experimental/complex_valued.py
lib/python3.9/site-packages/optax/_src/experimental/complex_valued_test.py
lib/python3.9/site-packages/optax/_src/experimental/extra_args.py
lib/python3.9/site-packages/optax/_src/experimental/extra_args_test.py
lib/python3.9/site-packages/optax/_src/factorized.py
lib/python3.9/site-packages/optax/_src/factorized_test.py
lib/python3.9/site-packages/optax/_src/float64_test.py
lib/python3.9/site-packages/optax/_src/linear_algebra.py
lib/python3.9/site-packages/optax/_src/linear_algebra_test.py
lib/python3.9/site-packages/optax/_src/lookahead.py
lib/python3.9/site-packages/optax/_src/lookahead_test.py
lib/python3.9/site-packages/optax/_src/loss.py
lib/python3.9/site-packages/optax/_src/loss_test.py
lib/python3.9/site-packages/optax/_src/numerics.py
lib/python3.9/site-packages/optax/_src/numerics_test.py
lib/python3.9/site-packages/optax/_src/privacy.py
lib/python3.9/site-packages/optax/_src/privacy_test.py
lib/python3.9/site-packages/optax/_src/schedule.py
lib/python3.9/site-packages/optax/_src/schedule_test.py
lib/python3.9/site-packages/optax/_src/second_order.py
lib/python3.9/site-packages/optax/_src/second_order_test.py
lib/python3.9/site-packages/optax/_src/stochastic_gradient_estimators.py
lib/python3.9/site-packages/optax/_src/stochastic_gradient_estimators_test.py
lib/python3.9/site-packages/optax/_src/test_utils.py
lib/python3.9/site-packages/optax/_src/transform.py
lib/python3.9/site-packages/optax/_src/transform_test.py
lib/python3.9/site-packages/optax/_src/update.py
lib/python3.9/site-packages/optax/_src/update_test.py
lib/python3.9/site-packages/optax/_src/utils.py
lib/python3.9/site-packages/optax/_src/utils_test.py
lib/python3.9/site-packages/optax/_src/wrappers.py
lib/python3.9/site-packages/optax/_src/wrappers_test.py
lib/python3.9/site-packages/optax/experimental/__init__.py
lib/python3.9/site-packages/optax/experimental/__pycache__/__init__.cpython-39.pyc
lib/python3.9/site-packages/optax/optax_test.py
