JAX version updates#151

Merged

lukebaumann merged 1 commit intomainfrom

lukebaumann/bump-jax

Jan 6, 2026

Commits on Jan 6, 2026

Included the latest stable version of JAX in GitHub actions. Bumped the requirment to jax>=0.7.0. Removed requirements.txt and rely exclusively on pyproject.toml.
lukebaumann
committed