AArthur Kaza·Sep 18Feeding the Beast: why your TPU is bored, and how to fix itA practical guide to data pipelines and profiling for JAX on TPU. Companion notebook: feeding_the_beast_jax_tpu.ipynb (Apache 2.0).4 min read·arthurkaza.com
