Tests

Equalize defaults #74

Job	Run time
lint	1m 31s
test (ubuntu-latest, 3.10, jax)	15m 1s
test (ubuntu-latest, 3.10, tensorflow)	11m 54s
test (ubuntu-latest, 3.10, torch)	5m 14s
test (ubuntu-latest, 3.11, jax)	15m 18s
test (ubuntu-latest, 3.11, tensorflow)	12m 30s
test (ubuntu-latest, 3.11, torch)	5m 24s
test (windows-latest, 3.10, jax)	15m 18s
test (windows-latest, 3.10, tensorflow)	15m 11s
test (windows-latest, 3.10, torch)	14m 29s
test (windows-latest, 3.11, jax)	15m 14s
test (windows-latest, 3.11, tensorflow)	15m 16s
test (windows-latest, 3.11, torch)	13m 1s
	2h 35m 21s

Provide feedback