├── .github └── workflows │ ├── python-publish.yml │ └── test.yml ├── .gitignore ├── LICENSE ├── README.md ├── grokfast.png ├── grokfast_pytorch ├── __init__.py └── grokfast.py ├── pyproject.toml └── tests └── test_grokfast_adam.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /grokfast.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/grokfast.png -------------------------------------------------------------------------------- /grokfast_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/grokfast_pytorch/__init__.py -------------------------------------------------------------------------------- /grokfast_pytorch/grokfast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/grokfast_pytorch/grokfast.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/test_grokfast_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/grokfast-pytorch/HEAD/tests/test_grokfast_adam.py --------------------------------------------------------------------------------