Tests

update Flow Matching (WIP) #22

Job	Run time
lint	1m 23s
test (ubuntu-latest, 3.10, jax)	3m 1s
test (ubuntu-latest, 3.10, tensorflow)	3m 17s
test (ubuntu-latest, 3.10, torch)	3m 16s
test (ubuntu-latest, 3.11, jax)	3m 16s
test (ubuntu-latest, 3.11, tensorflow)	3m 18s
test (ubuntu-latest, 3.11, torch)	3m 18s
test (windows-latest, 3.10, jax)	3m 6s
test (windows-latest, 3.10, tensorflow)	3m 2s
test (windows-latest, 3.10, torch)	3m 1s
test (windows-latest, 3.11, jax)	3m 1s
test (windows-latest, 3.11, tensorflow)	3m 4s
test (windows-latest, 3.11, torch)	3m 2s
	39m 5s

Provide feedback