https://github.com/helmholtz-analytics/heat

Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python

array-api data-analytics data-processing data-science distributed gpu hpc machine-learning massive-datasets mpi mpi4py multi-gpu multi-node-cluster numpy parallelism python pytorch tensors

Last synced: 15 May 2025

https://github.com/helmholtz-analytics/mpi4torch

An MPI wrapper for the pytorch tensor library that is automatically differentiable

Last synced: 08 Apr 2025

https://github.com/helmholtz-analytics/hpc_pytorch_loader

Efficient dataset storage in Python using memory-mapped arrays, Parquet, PyArrow IPC, and HDF5 to reduce inode usage

Last synced: 11 Oct 2025