├── .gitignore ├── LICENSE ├── README.md ├── kernels └── int8_matmul.py ├── pyproject.toml ├── ramtorch ├── __init__.py ├── helpers.py ├── modules │ ├── __init__.py │ └── linear.py ├── stochastic_optimizers │ ├── __init__.py │ └── adamw.py ├── zero1.py └── zero2.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | dist 3 | *.egg-info 4 | init.sh 5 | /.vscode 6 | 7 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/README.md -------------------------------------------------------------------------------- /kernels/int8_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/kernels/int8_matmul.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /ramtorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/ramtorch/__init__.py -------------------------------------------------------------------------------- /ramtorch/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/ramtorch/helpers.py -------------------------------------------------------------------------------- /ramtorch/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ramtorch/modules/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/ramtorch/modules/linear.py -------------------------------------------------------------------------------- /ramtorch/stochastic_optimizers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ramtorch/stochastic_optimizers/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/ramtorch/stochastic_optimizers/adamw.py -------------------------------------------------------------------------------- /ramtorch/zero1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/ramtorch/zero1.py -------------------------------------------------------------------------------- /ramtorch/zero2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/ramtorch/zero2.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lodestone-rock/RamTorch/HEAD/setup.py --------------------------------------------------------------------------------