TLDR AI · ENGINEERING & RESEARCH

HOW TO SCALE YOUR MODEL

jax-ml.github.io14 min read

Summary from the TLDR AI newsletter

This book discusses the science of scaling language models. It covers how TPUs and GPUs work, how they communicate with each other, how LLMs run on real hardware, and how to parallelize models during training and inference so they run efficiently at massive scale. The book answers questions about how expensive training a model should be, how much memory is needed to serve models, and more.