No project description provided
Project description
A library to experiment with new optimization algorithms in MLX.
import mlx_optimizers as optim
#... model, grads, etc.
optimizer = optim.DiffGrad(learning_rate=0.001)
optimizer.update(model, grads)
Coming to pip soon! :tada:
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mlx_optimizers-0.1.0.tar.gz
(23.3 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 mlx_optimizers-0.1.0.tar.gz.
File metadata
- Download URL: mlx_optimizers-0.1.0.tar.gz
- Upload date:
- Size: 23.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
e8c566f40cc88a4d15f0e6929b8df532af4b596d3d8444adc071dd2c953cdea2
|
|
| MD5 |
60cbd360fb205b29e28dd04b07535b24
|
|
| BLAKE2b-256 |
6d3747b7e1f851e65ff164630b5872514562a74dabf885a4b1f7e7859277bbcd
|
File details
Details for the file mlx_optimizers-0.1.0-py3-none-any.whl.
File metadata
- Download URL: mlx_optimizers-0.1.0-py3-none-any.whl
- Upload date:
- Size: 23.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7c3a1cc11549b828faa330679829d4082182d1888c33a50ad3a381309280ba2c
|
|
| MD5 |
1aefbb07996d297b01675727f87acf9f
|
|
| BLAKE2b-256 |
0f454008593540287bd7d9fb50d68a75394857e94901105ddcb8d43c0ee318aa
|