Student materials for DIS 2026 JAX-on-TPU workshop - View it on GitHub