A free, open textbook on engineering ML systems — building efficient, reliable AI from a single GPU up to warehouse-scale clusters. Goes beyond model design and MLOps tooling to the underlying science: scheduling, quantization, data pipelines, serving.
AI Train2023
Collection of runnable model implementations — LLaMA, Mistral, Stable Diffusion, Whisper, CLIP, plus LoRA fine-tuning — ported to the MLX array framework so they run natively on Apple silicon's unified memory rather than CUDA.
AI Infra2023
Provides a NumPy-like array framework for building and training ML on Apple silicon, with Python, C/C++, and Swift APIs plus PyTorch-style higher-level modules. Features lazy evaluation, composable AD/vectorization, and a unified-memory multi-device model so arrays can be used on CPU and GPU without explicit copies.
AI Image2023
Provides a diffusion-model studio for image, video, audio-video, editing, LoRA, and full training workflows so many model families share one inference and training framework.
AI Infra2023
Provides a PyTorch-native platform for experimenting with and scaling generative AI training, including composable parallelism, checkpointing, float8, logging, and Llama recipes.
AI Audio2024
Clones a voice from a 5-second sample for zero-shot TTS, or fine-tunes on ~1 minute of audio for few-shot synthesis. Covers Chinese, English, Japanese, Korean, and Cantonese, with a WebUI bundling vocal separation, ASR, and dataset labeling.
Embodied AI2024
A PyTorch-native, hardware-agnostic stack for robot learning: data collection, training, and deployment across 11+ robots, from SO100 to Unitree G1. Includes imitation, RL, and vision-language-action policies (ACT, Diffusion, Pi0, SmolVLA).
Chatbot2024
Creates personalized digital avatars (AI twins) by fine-tuning LLMs on users' chat history and binding them to chatbots. Provides an end-to-end pipeline — chat export, preprocessing with privacy filters, SFT/LoRA training, and deployment (Telegram/Discord/Slack). Best with larger models and substantial chat data.
AI Image2024
Produces real-time 3D reconstructions from multi-view images using Gaussian splatting, with on-device training and interactive viewing across native desktops, Android, and the browser. Uses WebGPU and the Burn ML framework to ship dependency-free binaries, a CLI, live training visualization, and streaming .ply support.
AI Dataset2024
Provides a multilingual, deduplicated corpus of public source code in Parquet for large-scale model training and evaluation. Includes license metadata, language splits, and streaming-friendly packaging for use with Hugging Face Datasets — suited to training code-focused foundation models but requires careful license/provenance review.
AI Video2024
Generates short videos from text, images, or videos and ships a full training/inference pipeline with checkpoints and demos. Key features include multi-stage training (VAE / 3D-VAE), rectified-flow training, video compression modules, and support for 2s–16s clips at up to 720p. Best for researchers and engineers who can provide substantial GPU resources.