Tests

Tests #2290

Job	Run time
Fast GPU tests Python 3.10 on ubuntu-22.04	6h 0m 13s
Python 3.10 on ubuntu-latest	1h 16m 10s
Fast tests Python 3.10 jax-default	11m 12s
Fast tests Python 3.10 jax-latest	11m 54s
Python 3.11 on ubuntu-latest	1h 23m 1s
Python 3.12 on ubuntu-latest	1h 39m 51s
Python 3.13 on ubuntu-latest	1h 16m 41s
Python 3.9 on macos-14	44m 44s
Python 3.10 on macos-15	58m 41s
	13h 42m 27s

Provide feedback