Stars
Spec-driven development (SDD) for AI coding assistants.
An educational resource to help anyone learn deep reinforcement learning.
AI agents running research on single-GPU nanochat training automatically
This repository contains the results and code for the MLPerf® Training v5.0 benchmark.
Reference implementations of MLPerf® training benchmarks
Run ML workloads seamlessly on cloud TPUs and GPUs with a single Python decorator. No infrastructure management required.
An open-source, code-first Python toolkit for building, evaluating, and deploying sophisticated AI agents with flexibility and control.
Minimal, lightweight JAX implementations of popular models.
Your own personal AI assistant. Any OS. Any Platform. The lobster way. 🦞
slime is an LLM post-training framework for RL Scaling.
A concise but complete full-attention transformer with a set of promising experimental features from various papers
Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.6, DeepSeek-V4, GLM-5.1, InternLM3, Llama4, ...) and 300+ MLLMs (Qwen3-VL, Qwen3-Omni, InternVL3.5, Ovis2.5, GLM4.5v, Gemma4, Llava, …
FlashMLA: Efficient Multi-head Latent Attention Kernels
torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytor…
Qwen-Image is a powerful image generation foundation model capable of complex text rendering and precise image editing.
TurboDiffusion: 100–200× Acceleration for Video Diffusion Models
pprof is a tool for visualization and analysis of profiling data
[ICLR & NeurIPS 2025] Repository for Show-o series, One Single Transformer to Unify Multimodal Understanding and Generation.
20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.
Making large AI models cheaper, faster and more accessible
Slide decks, coding exercises, and quick references for learning the JAX AI Stack
This repository contains implementations and illustrative code to accompany DeepMind publications