CMakeLists.txt
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
include/common.h
include/utils.h
include/adam_op/adam_op.h
include/adam_op/adam_op_impl_cpu.h
include/adam_op/adam_op_impl_cuda.cuh
src/CMakeLists.txt
src/extension.cpp
src/adam_op/adam_op.cpp
src/adam_op/adam_op_impl_cpu.cpp
src/adam_op/adam_op_impl_cuda.cu
torchopt/__init__.py
torchopt/base.py
torchopt/clip.py
torchopt/combine.py
torchopt/hook.py
torchopt/py.typed
torchopt/pytree.py
torchopt/typing.py
torchopt/update.py
torchopt/utils.py
torchopt/version.py
torchopt/visual.py
torchopt.egg-info/PKG-INFO
torchopt.egg-info/SOURCES.txt
torchopt.egg-info/dependency_links.txt
torchopt.egg-info/requires.txt
torchopt.egg-info/top_level.txt
torchopt/_C/adam_op.pyi
torchopt/accelerated_op/__init__.py
torchopt/accelerated_op/adam_op.py
torchopt/accelerated_op/_src/__init__.py
torchopt/accelerated_op/_src/adam_op.py
torchopt/alias/__init__.py
torchopt/alias/adam.py
torchopt/alias/adamw.py
torchopt/alias/rmsprop.py
torchopt/alias/sgd.py
torchopt/alias/utils.py
torchopt/diff/__init__.py
torchopt/diff/implicit/__init__.py
torchopt/diff/implicit/decorator.py
torchopt/diff/implicit/nn/__init__.py
torchopt/diff/implicit/nn/module.py
torchopt/diff/zero_order/__init__.py
torchopt/diff/zero_order/decorator.py
torchopt/distributed/__init__.py
torchopt/distributed/api.py
torchopt/distributed/autograd.py
torchopt/distributed/world.py
torchopt/linalg/__init__.py
torchopt/linalg/cg.py
torchopt/linalg/ns.py
torchopt/linalg/utils.py
torchopt/linear_solve/__init__.py
torchopt/linear_solve/cg.py
torchopt/linear_solve/inv.py
torchopt/linear_solve/normal_cg.py
torchopt/linear_solve/utils.py
torchopt/nn/__init__.py
torchopt/nn/module.py
torchopt/optim/__init__.py
torchopt/optim/adam.py
torchopt/optim/adamw.py
torchopt/optim/base.py
torchopt/optim/rmsprop.py
torchopt/optim/sgd.py
torchopt/optim/func/__init__.py
torchopt/optim/func/base.py
torchopt/optim/meta/__init__.py
torchopt/optim/meta/adam.py
torchopt/optim/meta/adamw.py
torchopt/optim/meta/base.py
torchopt/optim/meta/rmsprop.py
torchopt/optim/meta/sgd.py
torchopt/schedule/__init__.py
torchopt/schedule/polynomial.py
torchopt/transform/__init__.py
torchopt/transform/add_decayed_weights.py
torchopt/transform/nan_to_num.py
torchopt/transform/scale.py
torchopt/transform/scale_by_adam.py
torchopt/transform/scale_by_rms.py
torchopt/transform/scale_by_schedule.py
torchopt/transform/scale_by_stddev.py
torchopt/transform/trace.py
torchopt/transform/utils.py