AI Train2024
Predicts 3D structures of proteins, nucleic acids, and small-molecule complexes, the first fully open-source model to approach AlphaFold3 accuracy. Boltz-2 adds binding-affinity prediction that nears FEP simulation accuracy at ~1000x the speed.
AI Model2024
Generates high-quality, editable 3D assets from text or images and decodes to radiance fields, 3D Gaussians, or textured meshes. Ships pretrained models up to 2B parameters, a 500K asset dataset and training code; best used with image conditioning and a ≥16GB NVIDIA GPU.
AI Train2024
A 671B-parameter Mixture-of-Experts language model (37B activated) trained on 14.8T tokens with 128K context, FP8-first training, a Multi-Token Prediction module, and Hugging Face weights—focused on efficient MoE training and long-context use cases.
Embodied AI2024
Provides an open platform of omnimodal world models, datasets, and tools to build Physical AI — joint perception, generation, and action reasoning for robots, autonomous vehicles, and smart infrastructure. Supports images, video, audio, and action-conditioned workflows.
Provides end-to-end PyTorch scripts to download/prepare data, implement a transformer from scratch, train LLMs (13M→billion-scale) and generate text. Emphasizes educational clarity and single‑GPU experiments; useful for researchers or hobbyists, but large-scale training still requires substantial compute and engineering.
AI Train2025
Bundles a dataset, an interaction harness, and rubric-based reward functions into one RL environment for training and evaluating LLMs — also usable as an eval, synthetic-data pipeline, or agent harness for any OpenAI-compatible endpoint.
AI Infra2025
Spins up sandboxed VMs and containers (macOS, Linux, Windows, Android) that an AI agent can fully control through one unified SDK, cloud or local, plus a benchmark suite and background drivers that automate native apps without grabbing the cursor.
AI Infra2025
High-performance CUDA tensor-core GEMM kernel library for LLM workloads: supports FP8/FP4/BF16, fused Mega MoE and MQA scoring, and runtime JIT-compiled kernels. Targets NVIDIA SM90/SM100 and PyTorch—for teams working on low-level GPU kernel optimization.
AI Infra2025
Provides high-throughput, low-latency GPU communication kernels for Mixture-of-Experts (MoE) and expert-parallel workloads, with NVLink↔RDMA-aware forwarding, FP8/BF16 support, and low-latency RDMA hooks for inference decoding.
AI Train2025
An asynchronous, high-throughput framework for large-scale reinforcement learning and agentic training that scales to 1T+ MoE models and 1000+ GPUs, with native verifiers integration, end-to-end SFT/RL/evals, and Slurm/Kubernetes deployment; requires NVIDIA GPUs.
AI Train2025
inclusionAI (AReaL Team), Institute for Interdisciplinary Information Sciences, Tsinghua University +1Ant Group, Institute for Interdisciplinary Information Sciences, Tsinghua University
Trains LLM reasoning and agentic models with fully asynchronous reinforcement learning, decoupling rollout generation from policy updates for a 2.77x speedup over synchronous RL. Covers GRPO, PPO and DAPO across Megatron, FSDP, vLLM and SGLang backends.
AI Dataset2025
High-quality, efficiently verified and filtered web corpus for LLM pretraining — supplies ~1 trillion English tokens and ~120 billion Chinese tokens with English/Chinese Parquet splits. Designed for large-scale pretraining experiments and data-filtering research.