Implement support for pytree-valued differential equations (#833) #1511

Job	Run time
Run the quickstart with 3.13 on ubuntu-latest	22s
Run the quickstart with 3.12 on ubuntu-latest	32s
Run the quickstart with 3.11 on ubuntu-latest	24s
Dry-run benchmarks with 3.12 on ubuntu-latest	2m 13s
Lint with 3.12 on ubuntu-latest	48s
Test with 3.13 on ubuntu-latest	3m 5s
Test with 3.11 on ubuntu-latest	3m 3s
Build docs with 3.12 on ubuntu-latest	3m 8s
Test with 3.12 on ubuntu-latest	2m 59s
	16m 34s

Provide feedback