Make result containers JAX-compatible Equinox modules - #352

Open

mattlevine22 wants to merge 1 commit into

mainfrom

ml-dataclass-to-eqx

Commits on Sep 1, 2026

migrate dataclasses to equinox where possible for jax compat
mattlevine22
committed
6bcef75