Tests

Some love to model comparison #479

Job	Run time
lint	1m 22s
test (ubuntu-latest, 3.10, jax)	8m 5s
test (ubuntu-latest, 3.10, tensorflow)	21m 55s
test (ubuntu-latest, 3.10, torch)	10m 43s
test (ubuntu-latest, 3.11, jax)	8m 14s
test (ubuntu-latest, 3.11, tensorflow)	22m 46s
test (ubuntu-latest, 3.11, torch)	11m 2s
test (windows-latest, 3.10, jax)	15m 16s
test (windows-latest, 3.10, tensorflow)	33m 23s
test (windows-latest, 3.10, torch)	17m 21s
test (windows-latest, 3.11, jax)	15m 25s
test (windows-latest, 3.11, tensorflow)	39m 58s
test (windows-latest, 3.11, torch)	21m 28s
	3h 46m 58s

Provide feedback