add tests for summary networks [WIP] #351

Job	Run time
lint	1m 25s
test (ubuntu-latest, 3.10, jax)	11m 3s
test (ubuntu-latest, 3.10, tensorflow)	8m 45s
test (ubuntu-latest, 3.10, torch)	4m 50s
test (ubuntu-latest, 3.11, jax)	12m 25s
test (ubuntu-latest, 3.11, tensorflow)	8m 36s
test (ubuntu-latest, 3.11, torch)	5m 3s
test (windows-latest, 3.10, jax)	16m 23s
test (windows-latest, 3.10, tensorflow)	16m 26s
test (windows-latest, 3.10, torch)	11m 19s
test (windows-latest, 3.11, jax)	18m 55s
test (windows-latest, 3.11, tensorflow)	15m 14s
test (windows-latest, 3.11, torch)	11m 18s
	2h 21m 42s

Provide feedback