Allow `parallel_for` to take `chunk_size`