Tests

Complete implementation of Consistency Models #214

Job	Run time
lint	1m 23s
test (ubuntu-latest, 3.10, jax)	5m 37s
test (ubuntu-latest, 3.10, tensorflow)	5m 20s
test (ubuntu-latest, 3.10, torch)	3m 37s
test (ubuntu-latest, 3.11, jax)	6m 44s
test (ubuntu-latest, 3.11, tensorflow)	5m 37s
test (ubuntu-latest, 3.11, torch)	4m 0s
test (windows-latest, 3.10, jax)	11m 29s
test (windows-latest, 3.10, tensorflow)	11m 45s
test (windows-latest, 3.10, torch)	9m 42s
test (windows-latest, 3.11, jax)	11m 25s
test (windows-latest, 3.11, tensorflow)	12m 20s
test (windows-latest, 3.11, torch)	10m 49s
	1h 39m 48s

Provide feedback