Multi Gpu Python Packages

lilbee

The whole local AI stack in one executable: it runs and manages local AI models across every GPU, and it's a search engine you can talk to, with cited answers from your files, code, and the web. MCP server for coding agents, web crawler, TUI, CLI, REST API, Python library. No Ollama or LM Studio needed, works with both.

torchdr

TorchDR - PyTorch Dimensionality Reduction

deme3

A dual-GPU DEM solver with complex grain geometry support

deme

A dual-GPU DEM solver with complex grain geometry support

heat

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

tensortorrent

PyTorch compiler and Rust runtime for multi-CPU/GPU inference, NUMA-aware scheduling, model serving, parameter streaming, and activation spill.

supreme-unlearning

Multi-GPU framework for reproducible image-unlearning evaluation, with independent training, unlearning and evaluation seeds and extensible datasets, models, methods and metrics. Published at WIPE-OUT 2 (ECML-PKDD 2026).

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

Ray-powered accelerator for MinerU, turning PDF → Markdown into a scalable, cluster-ready data infrastructure. 基于 Ray 的 MinerU 加速层，将 PDF → Markdown 构建为可扩展、面向集群的数据基础设施。

cudamps

Python interface to CUDA Multi-Process Service