Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati...
by LightFog on Friday September 29, 2023 @02:29AM
(#37699994)
Porting existing codes is still a massive effort and there is low faith in long-term support from Google based software and hardware. I’m not aware on much (any) TPU use in scientific HPC.
[ Parent
| on HN ]
Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati...
by riedel on Friday September 29, 2023 @03:00AM
(#37700210)
There is not yet. But there is huge pressure on HPC Centers at least in Europe to also make resources available for ML. Already many scientific supercomputers have GPUs as accelerators. So we might have the other situation: HPC users are faced more and more with machines with spare accelerators and it will make sense to use them. Actually it would totally make sense if JAX development is in part also public also financed in this case (e.g. through EuroHPC).