Currently, only Python <3.14 can be used, due to constraints on the JAX version. These JAX versions have themselves caused some issues, e.g., for @cooijmanstim. We should allow for more modern JAX installations, which is currently blocked by the pinned version of cd_dynamax.
uv has an option to just ignore the pinned jax version of cd_dynamax, but this isn't ideal since it doesn't propagate to pip. A better option is to address this in cd_dynamax directly, which has an open PR: hd-UQ/cd_dynamax#12.
Currently, only Python <3.14 can be used, due to constraints on the JAX version. These JAX versions have themselves caused some issues, e.g., for @cooijmanstim. We should allow for more modern JAX installations, which is currently blocked by the pinned version of
cd_dynamax.uvhas an option to just ignore the pinned jax version ofcd_dynamax, but this isn't ideal since it doesn't propagate topip. A better option is to address this incd_dynamaxdirectly, which has an open PR: hd-UQ/cd_dynamax#12.