Library of JAX-based building blocks for reinforcement learning agents
An open source library for GPU-accelerated robot learning
A JAX research toolkit to build, edit, & visualize neural networks
A Model Context Protocol server for searching and analyzing arXiv
4M: Massively Multimodal Masked Modeling
This repository contains the official implementation of FastVLM
Refer and Ground Anything Anywhere at Any Granularity
Foundation Models for Time Series
Supercharge Your LLM with the Fastest KV Cache Layer
Set of tools to assess and improve LLM security
Tool for generating high quality Synthetic datasets
Agent toolkit providing semantic retrieval and editing capabilities
Open-source platform for building enterprise-grade agents
Fast3R: Towards 3D Reconstruction of 1000+ Images in One Forward Pass
MobileLLM Optimizing Sub-billion Parameter Language Models
DomainBed is a suite to test domain generalization algorithms
TorchMultimodal is a PyTorch library
ICLR2024 Spotlight: curation/training code, metadata, distribution
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)
Hackable and optimized Transformers building blocks
[CVPR 2025 Best Paper Award] VGGT
Code to accompany "A Method for Animating Children's Drawings"