Tests

merge upstream/dev into doc-polyversion #405

Job	Run time
lint	1m 18s
test (ubuntu-latest, 3.10, jax)	7m 41s
test (ubuntu-latest, 3.10, tensorflow)	8m 9s
test (ubuntu-latest, 3.10, torch)	5m 6s
test (ubuntu-latest, 3.11, jax)	8m 16s
test (ubuntu-latest, 3.11, tensorflow)	8m 25s
test (ubuntu-latest, 3.11, torch)	5m 46s
test (windows-latest, 3.10, jax)	13m 9s
test (windows-latest, 3.10, tensorflow)	14m 11s
test (windows-latest, 3.10, torch)	12m 46s
test (windows-latest, 3.11, jax)	14m 9s
test (windows-latest, 3.11, tensorflow)	15m 3s
test (windows-latest, 3.11, torch)	13m 7s
	2h 7m 6s

Provide feedback