Tests

Make linear regression the quickstart notebook #473

Job	Run time
lint	1m 29s
test (ubuntu-latest, 3.10, jax)	8m 7s
test (ubuntu-latest, 3.10, tensorflow)	21m 20s
test (ubuntu-latest, 3.10, torch)	15m 36s
test (ubuntu-latest, 3.11, jax)	8m 51s
test (ubuntu-latest, 3.11, tensorflow)	22m 33s
test (ubuntu-latest, 3.11, torch)	11m 40s
test (windows-latest, 3.10, jax)	15m 3s
test (windows-latest, 3.10, tensorflow)	31m 45s
test (windows-latest, 3.10, torch)	17m 28s
test (windows-latest, 3.11, jax)	15m 39s
test (windows-latest, 3.11, tensorflow)	32m 51s
test (windows-latest, 3.11, torch)	21m 3s
	3h 43m 25s

Provide feedback