Tests

Flexible implementation of point estimation #425

Job	Run time
lint	1m 31s
test (ubuntu-latest, 3.10, jax)	8m 30s
test (ubuntu-latest, 3.10, tensorflow)	9m 22s
test (ubuntu-latest, 3.10, torch)	6m 1s
test (ubuntu-latest, 3.11, jax)	9m 20s
test (ubuntu-latest, 3.11, tensorflow)	9m 39s
test (ubuntu-latest, 3.11, torch)	6m 37s
test (windows-latest, 3.10, jax)	14m 8s
test (windows-latest, 3.10, tensorflow)	16m 41s
test (windows-latest, 3.10, torch)	14m 32s
test (windows-latest, 3.11, jax)	15m 31s
test (windows-latest, 3.11, tensorflow)	14m 58s
test (windows-latest, 3.11, torch)	13m 59s
	2h 20m 49s

Provide feedback