Feature/jax production#1149

Merged

Jammy2211 merged 8 commits intofeature/jax_wrapperfrom

feature/jax_production

Jun 24, 2025

Commits on Jun 19, 2025

switchign use jax to config enables tests to pass

Jammy2211
authored and
Jammy2211
committed

Commits on Jun 20, 2025

simplify use jax import and make jax requirmeent

Jammy2211
authored and
Jammy2211
committed
disable warning when jax is used

Jammy2211
authored and
Jammy2211
committed
simplify factor jax use

Jammy2211
authored and
Jammy2211
committed
simplify jacobians

Jammy2211
authored and
Jammy2211
committed
move more jax imports

Jammy2211
authored and
Jammy2211
committed
fix pool and erfinv imports for cleaner JAX support

Jammy2211
authored and
Jammy2211
committed
all unit tests pass due to JAX fixes

Jammy2211
authored and
Jammy2211
committed