pyhpc-benchmarks

Tensors and Dynamic neural networks in Python with strong GPU acceleration

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

The fastai deep learning library

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Open3D: A Modern Library for 3D Data Processing

NumPy & SciPy for GPU

An open-source, low-code machine learning library in Python

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

cuDF - GPU DataFrame Library

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)