update README and TwoMoons example for discussion #167

Job	Run time
lint	1m 22s
test (ubuntu-latest, 3.10, jax)	5m 31s
test (ubuntu-latest, 3.10, tensorflow)	5m 26s
test (ubuntu-latest, 3.10, torch)	3m 52s
test (ubuntu-latest, 3.11, jax)	6m 21s
test (ubuntu-latest, 3.11, tensorflow)	5m 25s
test (ubuntu-latest, 3.11, torch)	4m 0s
test (windows-latest, 3.10, jax)	11m 2s
test (windows-latest, 3.10, tensorflow)	10m 13s
test (windows-latest, 3.10, torch)	12m 33s
test (windows-latest, 3.11, jax)	10m 35s
test (windows-latest, 3.11, tensorflow)	11m 36s
test (windows-latest, 3.11, torch)	10m 27s
	1h 38m 23s

Provide feedback