Fixing the dependency situation #1834

Job	Run time
python 3.9 - Lint	50s
python 3.9 - Type check	3m 9s
GPU Tests	3m 5s
python 3.9 - Build	54s
python 3.9 - Style	23s
python 3.9 - Docs	1m 36s
python 3.9 - Test	2m 55s
python 3.9 - Datasets integration	41s
python 3.9 - PyTorch integration	34s
python 3.9 - Transformers integration	1m 17s
python 3.9 - FairScale integration	26s
python 3.9 - W&B integration	3m 59s
python 3.9 - Beaker integration	3m 38s
python 3.9 - Flax integration	1m 41s
python 3.9 - GS integration	1m 5s
python 3.9 - Example - train_lm	1m 13s
python 3.8 - Test	2m 44s
python 3.10 - Test	2m 49s
Release	0s
	32m 59s

Provide feedback