2 ms·From Jax to VLIW: Tracing a Computation Through the TPU Compiler Stack1 points by EvgeniyZh 9mo ago