Merge pull request #426 from BastienZim/develop #47

Job	Run time
Build Docs	1m 36s
Lint code and check type hints	1m 15s
Run tests that use PyTorch (3.8) / run-tests	10m 46s
Run notebook tests (3.10) / run-tests	21m 7s
Run tests that use PyTorch (3.10) / run-tests	10m 54s
Run tests that use PyTorch (3.9) / run-tests	10m 40s
Run base tests (3.8) / run-tests	20m 22s
Run base tests (3.10) / run-tests	22m 24s
Run base tests (3.9) / run-tests	21m 23s
Run notebook tests (3.8) / run-tests	17m 36s
Run notebook tests (3.9) / run-tests	18m 54s
Push Docs and maybe release Package to TestPyPI	2m 3s
	2h 39m 0s

Provide feedback