How to Think About TPUs

https://jax-ml.github.io/scaling-book/tpus Part 2 of How To Scale Your Model (Part 1: Rooflines | Part 3: Sharding) This section is all about how TPUs work, how they're networked together to enable multi-chip training and inference,…

Executable Exports Symbols

There are actually several critical scenarios where an executable must export symbols. The confusion usually lies in the direction of the linking. You are right that Executable A rarely links…

Tailcall in AArch64

In AArch64 (ARM64), for a tail call to work, the current function must tear down its own stack frame before branching to the next function. If it didn't, the stack…