flextensor. FlexTensor is a tensor offloading and management library for PyTorch that enables running large models on limited GPU memory by intelligently offloading tensors between GPU and CPU memory.

github.com/ai-dynamo/flextensor

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.