[devetec]

Nope, probably not. It's using JAX, which works on both GPUs and TPUs.

[londons_explore]

All the constants in the design will be tuned to fit perfectly in TPU hardware dimensions.