Foundation Models for Time Series
Set of tools to assess and improve LLM security
MobileLLM Optimizing Sub-billion Parameter Language Models
A Production-ready Reinforcement Learning AI Agent Library
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
PyTorch3D is FAIR's library of reusable components for deep learning
Hackable and optimized Transformers building blocks
Official implementation of DreamCraft3D
Diffusion Transformer with Fine-Grained Chinese Understanding
NVIDIA Isaac GR00T N1.5 is the world's first open foundation model
AI-powered tool for developers, simplifying coding tasks
Scalable and user friendly neural forecasting algorithms.
A reactive notebook for Python
Machine Learning Pipelines for Kubeflow
OpenCompass is an LLM evaluation platform
The open-source data curation platform for LLMs
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction
End-to-End Library for Continual Learning based on PyTorch
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Test Suites for validating ML models & data
Build portable, production-ready MLOps pipelines
Implementation of DeepLabCut