MEMOPT: Universal Memory Fabric for AI Infrastructure. Open-sourced by Sophisticates.
-
Updated
Jun 4, 2026 - Python
MEMOPT: Universal Memory Fabric for AI Infrastructure. Open-sourced by Sophisticates.
A custom distributed training framework implementing ZeRO-1 and ZeRO-2 memory partitioning from scratch in PyTorch. Features async gradient hooks, sharded AdamW, and distributed checkpointing without relying on FSDP or HuggingFace Trainer.
Add a description, image, and links to the gpu-memory-management topic page so that developers can more easily learn about it.
To associate your repository with the gpu-memory-management topic, visit your repo's landing page and select "manage topics."