Tests

Diagnostics module #274

Job	Run time
lint	1m 19s
test (ubuntu-latest, 3.10, jax)	5m 45s
test (ubuntu-latest, 3.10, tensorflow)	5m 41s
test (ubuntu-latest, 3.10, torch)	4m 14s
test (ubuntu-latest, 3.11, jax)	6m 40s
test (ubuntu-latest, 3.11, tensorflow)	5m 47s
test (ubuntu-latest, 3.11, torch)	4m 19s
test (windows-latest, 3.10, jax)	11m 53s
test (windows-latest, 3.10, tensorflow)	12m 59s
test (windows-latest, 3.10, torch)	10m 4s
test (windows-latest, 3.11, jax)	11m 23s
test (windows-latest, 3.11, tensorflow)	13m 11s
test (windows-latest, 3.11, torch)	11m 31s
	1h 44m 46s

Provide feedback