CUDA Embedding Lookup Kernel Library - View it on GitHub