Are GPU embedding tables supported?

The code suggest it is but when running sync training with gpu there's a failure here because the referenced class does not exist.

https://github.com/bytedance/monolith/blob/88095e52234937cded0786f6dbb89f7731f9ec36/monolith/native_training/native_model.py#L877