If you want to experiment with these kinds of embeddings, a stabler version using Lorentz embeddings instead of Poincare was written by us at work for internal use.

https://github.com/theSage21/lorentz-embeddings