Tests

Flexible implementation of point estimation #475

Job	Run time
lint	1m 13s
test (ubuntu-latest, 3.10, jax)	9m 7s
test (ubuntu-latest, 3.10, tensorflow)	22m 38s
test (ubuntu-latest, 3.10, torch)	18m 52s
test (ubuntu-latest, 3.11, jax)	9m 21s
test (ubuntu-latest, 3.11, tensorflow)	25m 38s
test (ubuntu-latest, 3.11, torch)	19m 30s
test (windows-latest, 3.10, jax)	16m 25s
test (windows-latest, 3.10, tensorflow)	33m 7s
test (windows-latest, 3.10, torch)	22m 0s
test (windows-latest, 3.11, jax)	16m 17s
test (windows-latest, 3.11, tensorflow)	31m 43s
test (windows-latest, 3.11, torch)	23m 35s
	4h 9m 26s

Provide feedback