Tests

Diagnostics module #309

Job	Run time
lint	1m 29s
test (ubuntu-latest, 3.10, jax)	4m 25s
test (ubuntu-latest, 3.10, tensorflow)	4m 24s
test (ubuntu-latest, 3.10, torch)	4m 7s
test (ubuntu-latest, 3.11, jax)	4m 24s
test (ubuntu-latest, 3.11, tensorflow)	4m 23s
test (ubuntu-latest, 3.11, torch)	4m 30s
test (windows-latest, 3.10, jax)	4m 8s
test (windows-latest, 3.10, tensorflow)	4m 8s
test (windows-latest, 3.10, torch)	4m 9s
test (windows-latest, 3.11, jax)	4m 8s
test (windows-latest, 3.11, tensorflow)	4m 9s
test (windows-latest, 3.11, torch)	4m 9s
	52m 33s

Provide feedback