Memory Efficient PyTorch optimizers
Project description
FlashOptim
This is the official implementation of the FlashOptim: Optimizers for Memory Efficient Training by Databricks AI Research
Installation
pip install git+https://github.com/databricks/flashoptim.git
# or install it locally in editable mode for development
git clone https://github.com/databricks/flashoptim.git
cd flashoptim
pip install -e .
Usage
from flashoptim import FlashAdam, FlashSGD, FlashLion
optimizer = FlashAdam(model.parameters(), lr=1e-3)
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
flashoptim-0.0.1.tar.gz
(6.6 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file flashoptim-0.0.1.tar.gz.
File metadata
- Download URL: flashoptim-0.0.1.tar.gz
- Upload date:
- Size: 6.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.10.2 {"installer":{"name":"uv","version":"0.10.2","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9993aeb1febf0a4929579d9a900a08efe7f7521815007d35c64a8679aea95026
|
|
| MD5 |
35169af1bc15e5ed29b5c24b06186a93
|
|
| BLAKE2b-256 |
3b11ac1261aa4d0b4653674bf418b3b38933db3f4ecb065c9fce3b74a641898d
|
File details
Details for the file flashoptim-0.0.1-py3-none-any.whl.
File metadata
- Download URL: flashoptim-0.0.1-py3-none-any.whl
- Upload date:
- Size: 7.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.10.2 {"installer":{"name":"uv","version":"0.10.2","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
fbcadc4b460821112b41d46aba04d9aa75f98aff4da93a236803b07093cb688d
|
|
| MD5 |
a01a850a3cb0c79d148468bfa5eef40d
|
|
| BLAKE2b-256 |
e39e4fa59b279e04afee6d2a0d4bf34ca23470b75acdf2df8c0d42c82f119002
|
