GitHub repository: unixwzrd/TorchDevice
Primary language: Python
Created: 2024-09-18T16:37:45Z
Description: Module contains class TorchDevice abstracting CUDA/MPS/CPU, intercepting, and redirecting calls to the available device to assist in porting code from CUDA to MPS.