Tests

Merging dev with workflow #388

Job	Run time
lint	1m 15s
test (ubuntu-latest, 3.10, jax)	7m 45s
test (ubuntu-latest, 3.10, tensorflow)	8m 21s
test (ubuntu-latest, 3.10, torch)	5m 4s
test (ubuntu-latest, 3.11, jax)	8m 27s
test (ubuntu-latest, 3.11, tensorflow)	8m 29s
test (ubuntu-latest, 3.11, torch)	5m 25s
test (windows-latest, 3.10, jax)	14m 11s
test (windows-latest, 3.10, tensorflow)	15m 58s
test (windows-latest, 3.10, torch)	12m 27s
test (windows-latest, 3.11, jax)	14m 28s
test (windows-latest, 3.11, tensorflow)	15m 46s
test (windows-latest, 3.11, torch)	13m 10s
	2h 10m 46s

Provide feedback