I’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.