4 ms·

I am actually not surprised by JAX converting better to XLA. Also deep respect for anybody in this space as their is lot of complexity (?) to deal with at the f

I am actually not surprised by JAX converting better to XLA. Also deep respect for anybody in this space as their is lot of complexity (?) to deal with at the framework and compiler level.

felarof 2y ago
Thank you! Yeah, there are a few complexities and very little documentation around JAX, plus a lot of missing libraries.