Agent toolkit providing semantic retrieval and editing capabilities
Fast3R: Towards 3D Reconstruction of 1000+ Images in One Forward Pass
A Production-ready Reinforcement Learning AI Agent Library
PyTorch code and models for V-JEPA self-supervised learning from video
A PyTorch library for implementing flow matching algorithms
An implementation of a deep learning recommendation model (DLRM)
Self-supervised visual learning using momentum contrast in PyTorch
ImageBind One Embedding Space to Bind Them All
Hackable and optimized Transformers building blocks
[CVPR 2025 Best Paper Award] VGGT
Memory-efficient and performant finetuning of Mistral's models
tiktoken is a fast BPE tokeniser for use with OpenAI's models
Diffusion Transformer with Fine-Grained Chinese Understanding
NVIDIA Isaac GR00T N1.5 is the world's first open foundation model
A GPT-4o Level MLLM for Vision, Speech and Multimodal Live Streaming
AI-powered tool for developers, simplifying coding tasks
Machine Learning Pipelines for Kubeflow
Your open-source LLM evaluation toolkit
The open-source data curation platform for LLMs
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Test Suites for validating ML models & data
A comprehensive set of fairness metrics for datasets
High-level training, data augmentation, and utilities for Pytorch
JAX-based neural network library