Distributed Computing Python Packages

distributed

A distributed task scheduler for Dask

fugue

A unified interface for distributed computing. Fugue executes SQL, Python, Pandas, and Polars code on Spark, Dask and Ray without any rewrites.

torchrl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

fugue-sql-antlr

A unified interface for distributed computing. Fugue executes SQL, Python, Pandas, and Polars code on Spark, Dask and Ray without any rewrites.

daft

High-performance data engine for AI and multimodal workloads. Process images, audio, video, and structured data at any scale

pennylane-lightning

The Lightning plugin ecosystem provides fast quantum state-vector and tensor network simulators written in C++ for use with PennyLane.

ydf

A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Trees.

tdigest

t-Digest data structure in Python. Useful for percentiles and quantiles, including distributed enviroments like PySpark

dask-cloudprovider

Cloud provider cluster managers for Dask. Supports AWS, Google Cloud Azure and more...

dirac

DIRAC Grid

xoscar

Python actor framework for heterogeneous computing.

torchrl-nightly

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

pysail

Drop-in Apache Spark replacement written in Rust, unifying batch processing, stream processing, and compute-intensive AI workloads.

agi-env

Open-source platform for reproducible AI/ML workflows, from local experimentation to distributed workers and long-lived services.

backend-ai-manager

Backend.AI is a streamlined, container-based computing cluster platform that hosts popular computing/ML frameworks and diverse programming languages, with pluggable heterogeneous accelerator support including CUDA GPU, ROCm GPU, Gaudi NPU, Google TPU, GraphCore IPU and other NPUs.

backend-ai-agent

Backend.AI is a streamlined, container-based computing cluster platform that hosts popular computing/ML frameworks and diverse programming languages, with pluggable heterogeneous accelerator support including CUDA GPU, ROCm GPU, Gaudi NPU, Google TPU, GraphCore IPU and other NPUs.

agi-node

Open-source platform for reproducible AI/ML workflows, from local experimentation to distributed workers and long-lived services.

backend-ai-common

Backend.AI is a streamlined, container-based computing cluster platform that hosts popular computing/ML frameworks and diverse programming languages, with pluggable heterogeneous accelerator support including CUDA GPU, ROCm GPU, Gaudi NPU, Google TPU, GraphCore IPU and other NPUs.

backend-ai-client

Backend.AI is a streamlined, container-based computing cluster platform that hosts popular computing/ML frameworks and diverse programming languages, with pluggable heterogeneous accelerator support including CUDA GPU, ROCm GPU, Gaudi NPU, Google TPU, GraphCore IPU and other NPUs.

redis-dict

Python dictionary with Redis as backend, built for large datasets. Simplifies Redis operations for large-scale and distributed systems. Supports various data types, namespacing, pipelining, and expiration.

catalyst

Accelerated deep learning R&D

agi-cluster

Open-source platform for reproducible AI/ML workflows, from local experimentation to distributed workers and long-lived services.

spark-utils

Comfy Utilities for Spark Job Authoring

hadoop-fs-wrapper

Python Wrappers for Hadoop FileSystem