Tests

Add keras_kwargs utility #596

Job	Run time
test (ubuntu-latest, 3.10, jax)	1m 29s
test (ubuntu-latest, 3.10, numpy)	1m 30s
test (ubuntu-latest, 3.10, tensorflow)	1m 43s
test (ubuntu-latest, 3.10, torch)	1m 45s
test (ubuntu-latest, 3.11, jax)	1m 26s
test (ubuntu-latest, 3.11, numpy)	1m 42s
test (ubuntu-latest, 3.11, tensorflow)	1m 46s
test (ubuntu-latest, 3.11, torch)	1m 45s
test (windows-latest, 3.10, jax)	1m 31s
test (windows-latest, 3.10, numpy)	1m 30s
test (windows-latest, 3.10, tensorflow)	1m 31s
test (windows-latest, 3.10, torch)	1m 13s
test (windows-latest, 3.11, jax)	1m 29s
test (windows-latest, 3.11, numpy)	1m 30s
test (windows-latest, 3.11, tensorflow)	1m 30s
test (windows-latest, 3.11, torch)	1m 19s
	24m 39s

Provide feedback