feat(datasets) Add probability partitioner #11425

Job	Run time
Build, test and upload wheel	1m 51s
Strategy / FedMedian	1m 21s
Framework / bare	2m 40s
Strategy / FedTrimmedAvg	1m 10s
Framework / bare-https	1m 3s
Strategy / QFedAvg	1m 11s
Framework / bare-client-auth	50s
Strategy / FaultTolerantFedAvg	1m 16s
Framework / jax	1m 18s
Strategy / FedAvgM	1m 15s
Framework / pytorch	2m 16s
Strategy / FedAdam	1m 17s
Framework / tensorflow	3m 28s
Strategy / FedAdagrad	1m 20s
Framework / tabnet	2m 9s
Framework / opacus	2m 29s
Strategy / FedYogi	1m 17s
Framework / pytorch-lightning	2m 32s
Framework / scikit-learn	2m 51s
Framework / fastai	2m 15s
Framework / pandas	1m 11s
	37m 0s

Provide feedback