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

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 https://github.com/theSage21/lorentz-embeddings