An implementation of the hard sampling algorithm in NGAME: Negative Mining-aware Mini-batching for Extreme Classification https://arxiv.org/abs/2207.04452 implemented as a pytorch Dataset object.