Find rhymes with siamese network

I would need a pytorch implementation in order to find rhyming words based on this paper: https://aclanthology.org/W18-4509/ I parsed the database already and I also created the basic structure for the code. But currently we are not able to reproduce the results from the paper. Basically a siamese LSTM network needs to be implemented in pytorch.