3 ms·
But can Torch/TF/Flax do autodifferentiation on constants ordinary functions? No they cannot!
by macawfish 6y ago
But can Torch/TF/Flax do autodifferentiation on constants ordinary functions? No they cannot!
- komuher 6y agoFlax/jax can :)
- ChrisRackauckas 6y agoIf it's a pure function. Oh and if you have state-based control flow you have to turn off the JIT. Etc. If you take a standard library like some thermodynamics simulator and throw Jax on it do you expect it to work without modification? Most of the time it'll fail right at the start by using the wrong implementation of Numpy. So no, that's not "ordinary functions": those are functions where people consciously put in the effort to rewrite years of work onto Jax which is very different.