DEV: add option for jax backend - #12

Open

ColmTalbot wants to merge 3 commits into

Codecov / codecov/patch succeeded Jul 20, 2026 in 0s

78.08% of diff hit (target 0.00%)

View this Pull Request on Codecov

78.08% of diff hit (target 0.00%)

View more details on Codecov