Tests

rename diagnostics plots #376

Job	Run time
lint	1m 20s
test (ubuntu-latest, 3.10, jax)	8m 6s
test (ubuntu-latest, 3.10, tensorflow)	8m 15s
test (ubuntu-latest, 3.10, torch)	5m 8s
test (ubuntu-latest, 3.11, jax)	9m 13s
test (ubuntu-latest, 3.11, tensorflow)	8m 41s
test (ubuntu-latest, 3.11, torch)	5m 19s
test (windows-latest, 3.10, jax)	14m 50s
test (windows-latest, 3.10, tensorflow)	15m 48s
test (windows-latest, 3.10, torch)	11m 2s
test (windows-latest, 3.11, jax)	16m 39s
test (windows-latest, 3.11, tensorflow)	13m 48s
test (windows-latest, 3.11, torch)	11m 20s
	2h 9m 29s

Provide feedback