Jaret Burkett 609f3e5fa5 Fix issue with memory manager on MPS devices

2026-10-23 14:37:50 -06:00

..

__init__.py

Fix race condition that can corrupt grads under certain conditions.

2026-08-07 14:52:47 -06:00

manager_modules.py

Fixed issue with partial offloading torchao quants

2026-09-08 05:15:26 -06:00

manager.py

Fix issue with memory manager on MPS devices

2026-10-23 14:37:50 -06:00

test_memory_manager.py

Deepen offload prefetch pipeline with per-slot events

2026-06-07 16:07:13 -06:00