[londons_explore]

It's design is pretty specific to Google's TPU.

If you ran it elsewhere, you'd likely need to tweak the design or suffer quite a big performance penalty.

[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.