Cosmos-RL is a flexible and scalable Reinforcement Learning framework specialized for Physical AI applications.

torch models and primitives for birdseye view and voxel models for self driving tasks.

Development repository for the Triton language and compiler

Pipeline Parallelism for PyTorch

Instant neural graphics primitives: lightning fast NeRF and more

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

Pytorch domain library for recommendation systems

GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compression of numerical and other data types in HPC/ML applications.

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

An Open Source Machine Learning Framework for Everyone

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Open standard for machine learning interoperability

Caffe2 is a lightweight, modular, and scalable deep learning framework.