pylibraft#
2026-03-27
2 min read time
Applies to Linux
pylibraft is the Python interface to hipRAFT’s core algorithms and data
structures. It provides high-performance, GPU-accelerated primitives for common machine
learning and data analytics operations, including device resource management, distance
computations, random matrix generation, and sparse matrix utilities. pylibraft is
designed to integrate with the CUDA Array Interface(which has been extended to support
cupy and numba-hip) for interoperability with other GPU-based Python libraries.