Tests

Add free-form flows as inference networks #339

Job	Run time
lint	1m 32s
test (ubuntu-latest, 3.10, jax)	7m 56s
test (ubuntu-latest, 3.10, tensorflow)	7m 20s
test (ubuntu-latest, 3.10, torch)	5m 13s
test (ubuntu-latest, 3.11, jax)	9m 16s
test (ubuntu-latest, 3.11, tensorflow)	7m 25s
test (ubuntu-latest, 3.11, torch)	5m 33s
test (windows-latest, 3.10, jax)	13m 17s
test (windows-latest, 3.10, tensorflow)	14m 1s
test (windows-latest, 3.10, torch)	12m 19s
test (windows-latest, 3.11, jax)	15m 3s
test (windows-latest, 3.11, tensorflow)	14m 10s
test (windows-latest, 3.11, torch)	11m 16s
	2h 4m 21s

Provide feedback