Tests

Make linear regression the quickstart notebook #472

Job	Run time
lint	1m 37s
test (ubuntu-latest, 3.10, jax)	7m 59s
test (ubuntu-latest, 3.10, tensorflow)	21m 24s
test (ubuntu-latest, 3.10, torch)	10m 42s
test (ubuntu-latest, 3.11, jax)	8m 34s
test (ubuntu-latest, 3.11, tensorflow)	22m 54s
test (ubuntu-latest, 3.11, torch)	15m 59s
test (windows-latest, 3.10, jax)	14m 57s
test (windows-latest, 3.10, tensorflow)	32m 0s
test (windows-latest, 3.10, torch)	20m 58s
test (windows-latest, 3.11, jax)	15m 38s
test (windows-latest, 3.11, tensorflow)	31m 42s
test (windows-latest, 3.11, torch)	23m 3s
	3h 47m 27s

Provide feedback