← Back
Home › Systems › Frontier AI research

world-model

interactive generative world models

Genie 3, Cosmos

SystemTechniques · 6Primitives · 44 world-model Data loading pipelinesmachine-learning · medium: Turn files on disk into shuffled, sharded, resumable GPU batches fast enough that the accelerator never waits.›Near-duplicate detection with MinHa…machine-learning · medium: Estimate set Jaccard similarity from k min-hashes and find similar pairs with banded LSH buckets; the standard dedup for web crawls, training corpora and sequence collections.›Flow matchingml-training · medium: Training a velocity field along straight (rectified) paths from noise to data; sample by integrating an ODEKnowledge distillationml-training · medium: Train a small student to match a large teacher's softened outputs (plus the hard labels) so it keeps most of the teacher's accuracy at a fraction of the cost.›Variational autoencodersml-training · medium: VAEs and latent autoencoders: ELBO, KL weighting, latent compression for diffusion modelsVision transformerml-training · medium: ViT: patch embeddings, position encodings, encoder blocks; vision-language connectorsVector quantizationcompression · low: Codebook and pyramid vector quantization of coefficient vectorsAttention variantsmachine-learning · low: Multi-head, multi-query, grouped-query, sliding-window and latent attention; same softmax(QKᵀ/√d)V core, different trade-offs between quality and KV-cache size.›Transformer blockml-training · low: The decoder-only pre-norm block (attention plus MLP on a residual stream) behind every modern LLM, with exact shapes, parameter, FLOP and KV-cache counts.›Action tokenizationmachine-learning · low: Discretize or compress continuous robot actions into tokens a sequence model can predictPerceptual quality metrics (PSNR, S…media · low: Full-reference scores that say how bad a compressed picture looks: PSNR (pixel error), SSIM (structure), VMAF (learned fusion), and when each one lies.›Scene cut detectionmedia · low: Detect shot changes from frame-difference statisticsVideo codecsmedia · low: H.264/HEVC/AV1 basics: frames, GOPs, hardware decode/encode APIs and latency settingsClassifier free guidanceml-training · low: Mixing conditional and unconditional diffusion predictions with a guidance scale; condition dropout in trainingContext parallelismml-training · low: Splitting the sequence dimension across GPUs (ring attention, Ulysses all-to-all) for long-context trainingDiffusion modelsml-training · low: Train a network to undo Gaussian noise at every noise level, then generate by integrating from pure noise to data; the standard for images, video and audio.›Diffusion samplersml-training · low: DDPM/DDIM and ODE solvers (DPM-Solver, Euler, Heun) for few-step sampling from diffusion and flow modelsFlashAttentionml-training · low: Exact attention computed tile by tile in on-chip SRAM with online softmax, so the N×N score matrix never touches HBM; 2-4x faster, O(N) memory.›Mixed-precision trainingml-training · low: Run matmuls in bf16/fp16/fp8 on tensor cores while keeping master weights, optimizer state and reductions in fp32; about 2x speed, half the activation memory.›Paged KV cacheml-training · low: Store each sequence's attention keys/values in fixed-size blocks mapped through a block table, like virtual memory pages; near-zero fragmentation and copy-on-write sharing.›ZeRO and FSDP shardingml-training · low: Data parallelism that shards optimizer state, gradients and weights across ranks and gathers weights just in time; per-GPU model memory drops from 16Ψ to 16Ψ/N.›Object storagecloud · low: S3-compatible blob storage at cents per GB-month with 11 nines durability; pick by egress price, request price and latency, not by storage price alone.›+ 28 more primitives

Similar systems

Share the most building blocks with this one.