rlx-autodiff 0.2.8

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