A Debugging Journey into XLA, TPUs, and JAX
I published an account of an interesting* yak shaving journey I went on to debug a custom Pallas kernel on TPU. I’ll probably inline the content here at some point.
*Maybe to me and two other people on Earth