4 ms·
Nope, probably not. It's using JAX, which works on both GPUs and TPUs.
by devetec 5y ago
Nope, probably not. It's using JAX, which works on both GPUs and TPUs.
- londons_explore 5y agoAll the constants in the design will be tuned to fit perfectly in TPU hardware dimensions.