This website requires JavaScript.
Explore
Help
Sign In
github
/
ai-toolkit
Watch
1
Star
0
Fork
0
You've already forked ai-toolkit
Code
Issues
Pull Requests
Actions
1
Packages
Projects
Releases
Wiki
Activity
Files
2aa60e4ca50590c8a189311e341ae037736cbac8
ai-toolkit
/
toolkit
/
optimizers
History
Jaret Burkett
2aa60e4ca5
Update default agreement threshold for automagic v2 to be 0.5
2026-05-26 07:57:54 -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
Update default agreement threshold for automagic v2 to be 0.5
2026-05-26 07:57:54 -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