Jax – Why Is Everyone So Excited About This Framework
yash-sri.xyz
yash-sri.xyz
Frankly, if we were to change anything I would prefer to build on Candle, so if I need to drop down I can write Rust instead of C++. That would be a much bigger win than chasing after a pure functional dream land.
I used TF, Torch and Jax to implement a monte carlo sampling thingy and Jax was by far the most readable and performant.
I think the Cython use-case is kinda subsumed by the pypy jit right?