training word embeddings with XLNET

Question

I intend to train the word embeddings of all words in a corpus using XLNET and then compute their similarities. What package should I use? Are there any resources that could hel