search

marius-team / marius

Large scale graph learning on a single machine.
https://marius-project.org
Apache License 2.0
160 stars 45 forks source link

Compatibility with other GNN frameworks #139

Closed gamdwk closed 1 year ago

gamdwk commented 1 year ago

Can the marius framework be used with other GNN frameworks, such as dgl, pyg and so on?In addition, can I use my own model in marius?

JasonMoho commented 1 year ago

We do not support PyG and DGL as we use a different data model than these frameworks.

We also do not currently support general custom models without modification of the underlying C++. You can only use GraphSage and GAT layers as of now. In the future this may change, but Marius is more of a research system meant for efficiently scaling common GNN models and operators.

Ideally we would support the functionality you desire, but we are poor grad students without significant developer resources :)