Merge pull request #170 from Chase-Grajeda/deepset-refactor #591

Job	Run time
test (ubuntu-latest, 3.10, jax)	1m 21s
test (ubuntu-latest, 3.10, numpy)	1m 21s
test (ubuntu-latest, 3.10, tensorflow)	1m 33s
test (ubuntu-latest, 3.10, torch)	1m 21s
test (ubuntu-latest, 3.11, jax)	1m 21s
test (ubuntu-latest, 3.11, numpy)	1m 20s
test (ubuntu-latest, 3.11, tensorflow)	1m 19s
test (ubuntu-latest, 3.11, torch)	1m 14s
test (windows-latest, 3.10, jax)	1m 19s
test (windows-latest, 3.10, numpy)	1m 13s
test (windows-latest, 3.10, tensorflow)	1m 18s
test (windows-latest, 3.10, torch)	1m 18s
test (windows-latest, 3.11, jax)	1m 19s
test (windows-latest, 3.11, numpy)	1m 15s
test (windows-latest, 3.11, tensorflow)	1m 17s
test (windows-latest, 3.11, torch)	1m 17s
	21m 6s

Provide feedback