3 ms·
When I first read about JAX I thought it would kill Pytorch, but I'm not sure I can get on with an immutable language for tensor operations in deep learning. I
by jphoward 4y ago
When I first read about JAX I thought it would kill Pytorch, but I'm not sure I can get on with an immutable language for tensor operations in deep learning.
If I have an array `x` and want to set index 0 to 10, I cannot do:
x[0] = 10
I instead have to do:
y = x.at[0].set(10)
I'm sure I could get used to it, but it really puts me off.
- atorodius 4y agoAgreed that that's a bit ugly but at least in the ML context you rarely if ever need to do this (personally I only do this on the input to models, where we use pure numpy).
- uoaei 4y agoI feel the same. There are probably more ergonomic and generalizable ways to do whatever it is you need to do. Treat it as functional programming and let the XLA compiler handle things.
- jpeter 4y agoLooks like Java code. It's doa
- voz_ 4y agoI disagree.