2 ms·
It’s still all pytorch. Unless you’re working at Google, then maybe you use JAX.
by Legend2440 1y ago
It’s still all pytorch.
Unless you’re working at Google, then maybe you use JAX.
- mattnewton 1y agoJAX is quite popular in many labs outside of Google doing large scale training runs, because up until recently the parallelism ergonomics were way better. PyTorch core is catching up (maybe already witn the latest release, haven’t used it yet) and there are a lot of PyTorch using projects to study though.