Provides PyTorch code, pretrained checkpoints, and evaluation tooling for V-JEPA 2 — a Meta FAIR family of self-supervised video encoders and an action-conditioned world model. Includes training recipes, HuggingFace checkpoints, evaluation probes, and robot post‑training artifacts.
A PyTorch DTensor-native SPMD library for training and fine-tuning LLMs, VLMs, diffusion and retrieval models. Integrates with Hugging Face for day-0 model support, provides YAML-driven recipes, DTensor/FSDP2 parallelism and NVIDIA-optimized kernels (Transformer Engine, DeepEP, FlexAttn).
Enables bidirectional checkpoint conversion between Hugging Face and Megatron formats and provides a PyTorch-native training library with tensor/pipeline parallelism, FP8/BF16 mixed precision, SFT and PEFT (LoRA) support for large and multimodal models.
1,000,000 US-focused synthetic persona records (6M persona texts) grounded to demographic, geographic and personality distributions. Contains age, sex, education, occupation and ZCTA/city fields; CC BY 4.0 license for LLM training, diversity augmentation, and bias mitigation.
Trains and optimizes AI agents with reinforcement learning using almost zero code change. Works with any agent framework (LangChain, OpenAI Agents SDK, AutoGen, CrewAI) or none, and can selectively optimize a single agent inside a multi-agent system.
Provides a unified Python interface to collect data, train visual/dynamics world models, and evaluate them with model-predictive control across many standardized environments. Includes reference baselines, planning solvers, dataset converters, and LanceDB-backed formats for reproducible experiments. Best suited for researchers benchmarking world-model algorithms.
Forecasts financial candlesticks (OHLCV K-lines) with a decoder-only transformer pre-trained on 12B+ records from 45 exchanges. A tokenizer turns market data into discrete tokens, enabling price/volatility forecasting and synthetic K-line generation.
Self-supervised vision foundation model producing dense, patch-level features that transfer to classification, segmentation, depth, and detection with a frozen backbone. Spans ViT-S (21M) to ViT-7B (6.7B params), plus ConvNeXt and satellite variants.
A ~3.2M-conversation Hugging Face dataset of non-toxic human–ChatGPT interactions for instruction finetuning and evaluation; includes full transcripts plus request headers, hashed IP/geolocation, turn-level moderation scores and usage metadata.
A large multi-config collection of query–document pairs assembled to reproduce and extend the mGTE/LateOn data recipe for pre-training text embedding models. Data come in source-specific configs and include per-row drop/duplicate flags and guidance for using cleaned subsets for training.
Provides an NVFP4‑optimized training and inference infrastructure for long-form video diffusion models — supports multi-shot AR training, KV-cache and NVFP4 quantized inference, sequence-parallelism and async decoding for higher FPS and longer outputs.
Provides a Gymnasium-style API and tooling to create, deploy, and interact with isolated execution environments for agentic RL training. Includes async/sync clients, a web interface, CLI, Docker-based deployment, and Hugging Face Spaces integration.