https://www.tensorflow.org/recommenders/api_docs/python/tfrs/layers/embedding/TPUEmbedding?authuser=4
tfrs.layers.embedding.TPUEmbedding | TensorFlow Recommenders
A Keras layer for accelerating embedding lookups for large tables with TPU.
tfrslayersembeddingtensorflowrecommenders