Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/packquickly/schedule_free_optx
Schedule free optimiser implemented in JAX using Optimistix
https://github.com/packquickly/schedule_free_optx
Last synced: 16 days ago
JSON representation
Schedule free optimiser implemented in JAX using Optimistix
- Host: GitHub
- URL: https://github.com/packquickly/schedule_free_optx
- Owner: packquickly
- License: apache-2.0
- Created: 2024-05-29T13:17:39.000Z (6 months ago)
- Default Branch: main
- Last Pushed: 2024-05-29T13:25:34.000Z (6 months ago)
- Last Synced: 2024-10-25T21:51:58.342Z (23 days ago)
- Language: Python
- Size: 11.7 KB
- Stars: 14
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Schedule-free optimisers in JAX using Optimistix
An implementation of the new [schedule-free SGD and AdamW](https://arxiv.org/abs/2405.15682) optimisers in JAX using [Optimistix](https://github.com/patrick-kidger/optimistix).
# Installation
To install, simply download this repo and run
```pip install -e .```
in your Python environment. Then use `from schedule_free_optx import ScheduleFreeSGD, ScheduleFreeAdamW` and use them as any other Optimistix optimiser!