Note: this functionality will be somewhat limited until the shuffle / load logic in a forthcoming MR gets added