Python#

RapidsMPF provides a Python API for building high-performance multi-GPU data pipelines. The Python layer wraps the C++ core and integrates with popular distributed computing frameworks.

Quickstart#

API Reference#

  • Python API Reference — Full Python API reference (integrations, shuffler, communicator, memory, config)

Integrations#

The Python API includes ready-to-use integrations with:

  • Ray (rapidsmpf.integrations.ray) — use RapidsMPF within Ray tasks and actors.