I’m not very knowledgeable in this realm, so somebody clue me in. I always thought of JAX as targeted towards more “bespoke” stuff, is there any advantage to using it in a high-level way instead of Torch or TF? Anything in the performance or ecosystem etc?