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