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/version.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/_src/__init__.py
torchopt/_src/alias.py
torchopt/_src/base.py
torchopt/_src/clip.py
torchopt/_src/combine.py
torchopt/_src/hook.py
torchopt/_src/implicit_diff.py
torchopt/_src/linalg.py
torchopt/_src/linear_solve.py
torchopt/_src/schedule.py
torchopt/_src/transform.py
torchopt/_src/typing.py
torchopt/_src/update.py
torchopt/_src/utils.py
torchopt/_src/visual.py
torchopt/_src/accelerated_op/__init__.py
torchopt/_src/accelerated_op/adam_op.py
torchopt/_src/optimizer/__init__.py
torchopt/_src/optimizer/adam.py
torchopt/_src/optimizer/adamw.py
torchopt/_src/optimizer/base.py
torchopt/_src/optimizer/rmsprop.py
torchopt/_src/optimizer/sgd.py
torchopt/_src/optimizer/func/__init__.py
torchopt/_src/optimizer/func/base.py
torchopt/_src/optimizer/meta/__init__.py
torchopt/_src/optimizer/meta/adam.py
torchopt/_src/optimizer/meta/adamw.py
torchopt/_src/optimizer/meta/base.py
torchopt/_src/optimizer/meta/rmsprop.py
torchopt/_src/optimizer/meta/sgd.py