Skip to main content

Mochi Lab

Innovative Research & Development

research

Research

Original research papers and in-depth surveys spanning continual learning, world models, diffusion methods, and more.

programming

Systems & Programming

Hands-on guides to GPU programming with CUDA and Triton, PyTorch internals, x86 assembly, distributed training, and ML infrastructure.

math

Math & Theory

Mathematical foundations for machine learning — linear algebra, optimization, probability, information theory, and the math behind transformers and diffusion models.