3 ms·
Jax – Why Is Everyone So Excited About This Framework
- mountainriver 2y agoI’m still not sold that we need a new framework in this space. A lot of these articles talk about the amazing features of Jax but not the sharp edges it has. 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.
- 3np 2y agoJust cursory looking at Jax, it does feel like >95% of the "everyone excited about Jax" people would get >95% of the performance win for 5% of the increased complexity by using Cython instead...
- ysleepy 2y agoDoes Cython run on the GPU? 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?
- evjrob 2y agoThis blog seems to be a pretty blatant rip off of the autodidax page in the JAX documentation: https://jax.readthedocs.io/en/latest/autodidax.html https://jax.readthedocs.io/en/latest/autodidax.html
- ein0p 2y ago“Everyone is excited” but the share of papers that use it hasn’t budged from 3% in years.
- pjmlp 2y agoEveryone that happens to be a Google employee, I guess.
- CRConrad 2y agoQuestion assumes facts not in evidence: I'm not the least excited about it.