Add more tests for Jax backend

Created by: gcarleo

We are a little short of tests for the jax backend

added help wanted v3.0 labels

changed milestone to %Version 3.0

Created by: jamesETsmith

Hi @gcarleo, I'm happy to help with this. I assume I should start with branch v3.0?

Created by: gcarleo

Hey @jamesETsmith , yes v3.0 is the one! I would maybe wait for a couple of days though, so that we stabilize a few things on the jax side

Created by: gcarleo

It looks like a lot of new tests have already been added by @PhilipVinc while we has fighting to fix a few issues with Jax objects, so we might as well close this one too!

closed