GitHub repository: shssf/heat
Description: Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python