Tests

Working version of couplings with residual internal networks #544

Job	Run time
test (ubuntu-latest, 3.10, jax)	15s
test (ubuntu-latest, 3.10, numpy)	13s
test (ubuntu-latest, 3.10, tensorflow)	13s
test (ubuntu-latest, 3.10, torch)	13s
test (ubuntu-latest, 3.11, jax)	14s
test (ubuntu-latest, 3.11, numpy)	15s
test (ubuntu-latest, 3.11, tensorflow)	12s
test (ubuntu-latest, 3.11, torch)	9s
test (windows-latest, 3.10, jax)	13s
test (windows-latest, 3.10, numpy)	13s
test (windows-latest, 3.10, tensorflow)	14s
test (windows-latest, 3.10, torch)	10s
test (windows-latest, 3.11, jax)	12s
test (windows-latest, 3.11, numpy)	12s
test (windows-latest, 3.11, tensorflow)	12s
test (windows-latest, 3.11, torch)	12s
	3m 22s

Provide feedback