Tests

Tests #2274

Job	Run time
Fast GPU tests Python 3.10 on ubuntu-22.04	2h 15m 30s
Python 3.10 on ubuntu-latest	1h 16m 59s
Python 3.11 on ubuntu-latest	1h 14m 51s
Fast tests Python 3.10 jax-default	10m 34s
Python 3.12 on ubuntu-latest	1h 23m 36s
Fast tests Python 3.10 jax-latest	10m 48s
Python 3.13 on ubuntu-latest	1h 18m 0s
Python 3.9 on macos-13	1h 17m 10s
Python 3.10 on macos-14	1h 6m 58s
	10h 14m 26s

Provide feedback