Need help with torchkge?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

225 Stars 29 Forks Other 752 Commits 8 Opened issues


TorchKGE: Knowledge Graph embedding in Python and PyTorch.

Services available


Need anything else?

Contributors list



.. image:: :align: right :width: 100px :alt: logo torchkge

.. image:: :target:

.. image:: :target:

.. image:: :target: :alt: Documentation Status

.. image:: :target: :alt: Updates

TorchKGE: Knowledge Graph embedding in Python and Pytorch.

TorchKGE is a Python module for knowledge graph (KG) embedding relying solely on Pytorch. This package provides researchers and engineers with a clean and efficient API to design and test new models. It features a KG data structure, simple model interfaces and modules for negative sampling and model evaluation. Its main strength is a highly efficient evaluation module for the link prediction task, a central application of KG embedding. It has been

_ to be up to five times faster than
_ and twenty-four times faster than
_. Various KG embedding models are also already implemented. Special attention has been paid to code efficiency and simplicity, documentation and API consistency. It is distributed using PyPI under BSD license.


If you find this code useful in your research, please consider citing our

_ (presented at
_ 2020):

.. code-block::

    title={TorchKGE: Knowledge Graph Embedding in Python and PyTorch},
    author={Armand Boschin},
    booktitle={International Workshop on Knowledge Graph: Mining Knowledge Graph for Deep Insights},
  • Free software: BSD license
  • Documentation:

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.