This website requires JavaScript.
Explore
Help
Register
Sign In
ostris
/
ai-toolkit
Watch
1
Star
0
Fork
0
You've already forked ai-toolkit
mirror of
https://github.com/ostris/ai-toolkit.git
synced
2026-05-11 16:30:40 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
hidream_o1
Add File
New File
Upload File
Apply Patch
ai-toolkit
/
toolkit
/
optimizers
History
Jaret Burkett
6bb8acbffc
Add agreement_threshold default of 0.6 to automagic 2
2026-05-05 19:13:00 -06:00
..
adafactor.py
Added a basic torch profiler that can be used in config during development to find some obvious issues.
2025-06-17 13:03:39 -06:00
adam8bit.py
Various bug fixes and optimizations for quantized training. Added untested custom adam8bit optimizer. Did some work on LoRM (dont use)
2024-11-20 09:16:55 -07:00
automagic2.py
Add agreement_threshold default of 0.6 to automagic 2
2026-05-05 19:13:00 -06:00
automagic.py
Reworked automagic optimizer and did more testing. Starting to really like it. Working well.
2025-04-28 08:01:10 -06:00
optimizer_utils.py
Added a basic torch profiler that can be used in config during development to find some obvious issues.
2025-06-17 13:03:39 -06:00
prodigy_8bit.py
Added adafactor implementation that handles stochastic rounding of update and accumulation
2024-10-30 05:25:57 -06:00
test_optimizers.py
Add optimizer test suite and make minor speed adjustments to Automagicv2
2026-05-05 10:02:30 -06:00