A JAX-native LLM Post-Training Library
Flax is a neural network library for JAX
A library to generate LaTeX expression from Python code
Library of JAX-based building blocks for reinforcement learning agents
A JAX research toolkit to build, edit, & visualize neural networks
A secure sandbox environment for malware developers and red teamers
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
Tool for generating high quality Synthetic datasets
TorchMultimodal is a PyTorch library
ICLR2024 Spotlight: curation/training code, metadata, distribution
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)
[CVPR 2025 Best Paper Award] VGGT
PyTorch code and models for the DINOv2 self-supervised learning
Memory-efficient and performant finetuning of Mistral's models
Official implementation of DreamCraft3D
A lightweight data processing framework built on DuckDB and 3FS
DeepEP: an efficient expert-parallel communication library
Research code artifacts for Code World Model (CWM)
The Missing Semester of Your CS Education
Diffusion Transformer with Fine-Grained Chinese Understanding