Tests

Fix single parameter plots (#303) #469

Job	Run time
lint	1m 20s
test (ubuntu-latest, 3.10, jax)	8m 22s
test (ubuntu-latest, 3.10, tensorflow)	21m 56s
test (ubuntu-latest, 3.10, torch)	15m 20s
test (ubuntu-latest, 3.11, jax)	9m 10s
test (ubuntu-latest, 3.11, tensorflow)	23m 18s
test (ubuntu-latest, 3.11, torch)	11m 35s
test (windows-latest, 3.10, jax)	13m 46s
test (windows-latest, 3.10, tensorflow)	31m 34s
test (windows-latest, 3.10, torch)	20m 15s
test (windows-latest, 3.11, jax)	13m 22s
test (windows-latest, 3.11, tensorflow)	31m 21s
test (windows-latest, 3.11, torch)	19m 1s
	3h 40m 20s

Provide feedback