20 posts in total

---

2026

02-26

XLA02 - shapes, layout & tiling

02-25

XLA01 - architecture & workflows

02-22

Knowledge Distillation 101

02-19

GPU mode - lecture2 - CUDA 101

02-19

Pallas 101 - multi-backend kernel for JAX

02-07

5D parallelism in LLM training

01-25

Memory usage breakdown during Training

2025

12-22

JAX 101

12-05

Jeff Dean & Gemini team QA at NeurIPS ‘25

11-11

Pytorch Conference & Ray Summit 2025 summary