Add support for more complex parallelism strategies to benchmark_litgpt.py #311

Job	Run time
check-package / init-store	3s
check-schema / schema	11s
precommit-run / pre-commit	32s
check-package / pkg-build (ubuntu-latest)	25s
check-package / pkg-check (ubuntu-latest, 3.10)	1m 17s
check-package / pkg-check (ubuntu-latest, 3.11)	1m 22s
check-package / pkg-check (macOS-latest, 3.10)	1m 37s
check-package / pkg-check (macOS-latest, 3.11)	1m 54s
check-package / pkg-check (windows-latest, 3.10)	1m 49s
check-package / pkg-check (windows-latest, 3.11)	1m 49s
check-package / pkg-guardian	1m 14s
	12m 13s

Provide feedback