From Jax to VLIW: Tracing a Computation Through the TPU Compiler Stack

10 pointsposted a month ago
by mario1870

4 Comments

peter_d_sherman

a month ago

This is by far the absolute best "how things go from a high-level (AI/LLM/Inference/Python/JAX/etc.) to low-level (TPU equivalent of assembly language instructions)" article that I've read to date!

Upvoted and favorited!

A keeper -- for present and future study...

Well done!

patrick_toulme

a month ago

Thank you. I am the OP. Are there any other articles you would like me to cover?

t-k-233

a month ago

Seems that the page and the linkedin post are pulled down... :(

patrick_toulme

a month ago

It is back up now. I am the OP and I have reposted it