CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
configs/base.yaml
configs/gpt2_medium.yaml
configs/gpt2_small.yaml
configs/gpt_small.yaml
configs/llama_1b.yaml
configs/vit_small.yaml
scao/__init__.py
scao/distributed.py
scao/logging.py
scao/optimizer.py
scao/preconditioner.py
scao/py.typed
scao/utils.py
scao.egg-info/PKG-INFO
scao.egg-info/SOURCES.txt
scao.egg-info/dependency_links.txt
scao.egg-info/requires.txt
scao.egg-info/top_level.txt
scao/cuda/__init__.py
scao/cuda/setup.py
scao/integrations/__init__.py
scao/integrations/huggingface.py