5 ms·
> I'm not convinced the same won't happen to JAX And now there are already multiple NN libraries for JAX from Google...
by iflp 6y ago
> I'm not convinced the same won't happen to JAX
And now there are already multiple NN libraries for JAX from Google...
- joaogui1 6y agoThere are a bunch of frameworks built on top of Pytorch too (fastAI, lighting, torchbearer, ignite...), I don't see why this should be a problem (or at least a problem to JAX but not to Pytorch)
- MiroF 6y agoIMO, this is not a fair comparison because Pytorch spans a larger amount of abstraction than jax (I don't quite know how to explain it other than "spans a larger amount of abstraction"). You can do much of the jax stuff in pytorch, you can't do the high level nn.LSTM stuff in jax, you have to use like flax or objax or something.