Support specifying a torch range #3843

Job	Run time
linux (linux.12xlarge, cpu) / linux-job	4m 16s
macos (macos-m1-stable) / macos-job	3m 7s
linux (linux.g5.4xlarge.nvidia.gpu, cuda, 11.8) / linux-job	9m 25s
windows (windows.4xlarge, cpu) / windows-job	15m 58s
windows (windows.g5.4xlarge.nvidia.gpu, cuda, 11.8) / windows-job	13m 9s
	45m 55s

Provide feedback