Development repository for the Triton language and compiler

PyTorch native quantization and sparsity for training and inference

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Apache Beam is a unified programming model for Batch and Streaming data processing.

Protocol Buffers - Google's data interchange format

Reference implementations of MLPerf® training benchmarks

An Open Source Machine Learning Framework for Everyone