[Performance] Faster slice sampler (#2031) #1170

Job	Run time
unittests-pettingzoo / linux-job	9m 9s
unittests-atari-dqn (3.9, 12.1) / linux-job	12m 28s
unittests-brax (3.9, 12.1) / linux-job	9m 2s
unittests-d4rl (3.9, 12.1) / linux-job	8m 39s
unittests-envpool (3.9, 12.1) / linux-job	8m 40s
unittests-gendgrl (3.9, 12.1) / linux-job	6m 53s
unittests-gym (3.9, 12.1) / linux-job	18m 33s
unittests-jumanji (3.9, 12.1) / linux-job	8m 24s
unittests-minari (3.9, 12.1) / linux-job	16m 49s
unittests-openx (3.9, 12.1) / linux-job	25m 19s
unittests-robohive (3.9, 12.1) / linux-job	8m 58s
unittests-sklearn (3.9, 12.1) / linux-job	6m 0s
unittests-roboset (3.9, 12.1) / linux-job	10m 0s
unittests-smacv2 (3.9, 12.1) / linux-job	18m 35s
unittests-vd4rl (3.9, 12.1) / linux-job	8m 24s
unittests-vmas (3.9, 12.1) / linux-job	11m 39s
	3h 7m 32s

Provide feedback