Re:TPU v4 provides exaFLOPS-scale ML with industry-leading e...
by nestorD on Wednesday April 05, 2023 @11:23PM
(#35463765)
> model compatibility
JAX will compile your model with XLA today, letting you run it on Nvidia, Apple Silicon (CPU for now), google's TPU, etc.
Overall, XLA really shines on TPUs, is doing very well on GPUs and sadly lags behind on CPUs (reaching about single core performance).
As a bonus, JAX is basically Numpy++: you can use it to build scientific/numerical software and run it easily on GPU and TPU!
Re:TPU v4 provides exaFLOPS-scale ML with industry-leading e...
by gallabytes on Thursday April 06, 2023 @12:08AM
(#35464016)
yeah we trained v5 on TPUs and continue to train on them.
That said, it's also incredible how fast things move in this space:
> Midjourney, one of the leading text-to-image AI startups, have been using Cloud TPU v4 to train their state-of-the-art model, coincidentally also called “version four”.
Midjourney is already on v5 as of the date of publication of this press release.