Example Notebook for Users of 1.0 moving to 2.0 #506

Job	Run time
lint	1m 9s
test (ubuntu-latest, 3.10, jax)	7m 19s
test (ubuntu-latest, 3.10, tensorflow)	17m 25s
test (ubuntu-latest, 3.10, torch)	8m 28s
test (ubuntu-latest, 3.11, jax)	7m 29s
test (ubuntu-latest, 3.11, tensorflow)	19m 0s
test (ubuntu-latest, 3.11, torch)	8m 46s
test (windows-latest, 3.10, jax)	10m 57s
test (windows-latest, 3.10, tensorflow)	27m 47s
test (windows-latest, 3.10, torch)	16m 55s
test (windows-latest, 3.11, jax)	12m 42s
test (windows-latest, 3.11, tensorflow)	30m 8s
test (windows-latest, 3.11, torch)	18m 56s
	3h 7m 1s

Provide feedback