Fix for XLML tests failing with jax nightly#242

Merged

entrpn merged 1 commit intomainfrom

rbierneni-fix-xlml-test

Sep 8, 2025

Commits on Sep 5, 2025

Do unboxing only on concrete states and not abstract_state
Rohan-Bierneni
committed