Tests

Revert clamp changes from aeee627 (destabilized training) #332

Job	Run time
lint	1m 15s
test (ubuntu-latest, 3.10, jax)	5m 4s
test (ubuntu-latest, 3.10, tensorflow)	5m 4s
test (ubuntu-latest, 3.10, torch)	4m 48s
test (ubuntu-latest, 3.11, jax)	5m 4s
test (ubuntu-latest, 3.11, tensorflow)	5m 5s
test (ubuntu-latest, 3.11, torch)	4m 49s
test (windows-latest, 3.10, jax)	4m 49s
test (windows-latest, 3.10, tensorflow)	4m 48s
test (windows-latest, 3.10, torch)	4m 50s
test (windows-latest, 3.11, jax)	5m 3s
test (windows-latest, 3.11, tensorflow)	4m 48s
test (windows-latest, 3.11, torch)	4m 48s
	1h 0m 15s

Provide feedback