flexi-hash-embedding (pypi): PyTorch Extension Library of Optimized Scatter Operations
Keywords: pytorch|scatter|groupby|embedding|hashing|variable|fixed