mrq bd0a36ba8d I swear I keep seeing tqdm flicker back a number		2024-05-10 18:36:01 -05:00
..
__init__.py
distributed.py	simple DDP wrapper (for my NVlink test)	2024-05-04 11:48:26 -05:00
sampler.py
trainer.py	I swear I keep seeing tqdm flicker back a number	2024-05-10 18:36:01 -05:00
utils.py	nasty hotfix for transformer's Mixtral throwing an error when batch sizes > 1	2024-01-26 19:41:12 -06:00
wrapper.py	crammed in DAdaptation (doesn't seem worth it) and ScheduleFree (forgot I wanted to weeks ago, seems promising), optimization wrapper cleanup, test trainer changes, etc.	2024-05-09 20:28:20 -05:00