Multi Gpu Python Packages

torchdr

TorchDR - PyTorch Dimensionality Reduction

heat

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

gradient-accumulator

:dart: Gradient Accumulation for TensorFlow 2

tensordiffeq

Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed computing

flash-mineru

Fast Inference Architecture for MinerU

cudamps

Python interface to CUDA Multi-Process Service