rlx-autodiff 0.2.3

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