rlx-autodiff 0.2.9

JAX-shaped transforms for RLX MIR — autodiff, JVP/HVP, vmap
Documentation