Jaret Burkett 989ebfaa11 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

..

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

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