🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

An Open Source Machine Learning Framework for Everyone

The NVIDIA GPU driver container allows the provisioning of the NVIDIA driver through the use of containers.

SGLang is a fast serving framework for large language models and vision language models.

Kimi K2 is the large language model series developed by Moonshot AI team

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Build and run containers leveraging NVIDIA GPUs

A generative world for general-purpose robotics & embodied AI learning.

Protocol Buffers - Google's data interchange format

🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning

Multipass orchestrates virtual Ubuntu instances

TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. Tensor…

A PyTorch native platform for training generative AI models

GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.

A high-throughput and memory-efficient inference and serving engine for LLMs

scikit-learn: machine learning in Python

MinIO is a high-performance, S3 compatible object store, open sourced under GNU AGPLv3 license.

A Python framework for accelerated simulation, data generation and spatial computing.

openpilot is an operating system for robotics. Currently, it upgrades the driver assistance system on 300+ supported cars.

slime is an LLM post-training framework for RL Scaling.

Universal Robots ROS2 driver supporting CB3 and e-Series

Open sourced digital twin simulator for Autoware

verl: Volcano Engine Reinforcement Learning for LLMs

Open3D: A Modern Library for 3D Data Processing

Unified framework for robot learning built on NVIDIA Isaac Sim

Open-source framework for the research and development of foundation models.

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

An extremely fast Python package and project manager, written in Rust.