A Python library transfers PyTorch tensors between CPU and NVMe
-
Updated
Apr 27, 2023 - C++
A Python library transfers PyTorch tensors between CPU and NVMe
Add a description, image, and links to the colossal-ai topic page so that developers can more easily learn about it.
To associate your repository with the colossal-ai topic, visit your repo's landing page and select "manage topics."