Do you mean as an additional sampler into the PyTorch code base?
If so, please create a feature request with the description of the use case and your changes on GitHub 2, where it can be discussed further.