MilaGraph
Research group led by Prof. Jian Tang at Mila-Quebec AI Institute (https://mila.quebec/) focusing on graph representation learning and graph neural networks.
Montreal
Pinned Repositories
ConfGF
Implementation of Learning Gradient Fields for Molecular Conformation Generation (ICML 2021).
GearNet
GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)
GMNN
Graph Markov Neural Networks
graphvite
GraphVite: A General and High-performance Graph Embedding System
KnowledgeGraphEmbedding
LiteratureDL4Graph
A comprehensive collection of recent papers on graph deep learning
NBFNet
Official implementation of Neural Bellman-Ford Networks (NeurIPS 2021)
RecommenderSystems
torchdrug
A powerful and flexible machine learning platform for drug discovery
ULTRA
A foundation model for knowledge graph reasoning
MilaGraph's Repositories
DeepGraphLearning/torchdrug
A powerful and flexible machine learning platform for drug discovery
DeepGraphLearning/KnowledgeGraphEmbedding
DeepGraphLearning/graphvite
GraphVite: A General and High-performance Graph Embedding System
DeepGraphLearning/ULTRA
A foundation model for knowledge graph reasoning
DeepGraphLearning/GearNet
GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)
DeepGraphLearning/NBFNet
Official implementation of Neural Bellman-Ford Networks (NeurIPS 2021)
DeepGraphLearning/AStarNet
Official implementation of A* Networks
DeepGraphLearning/GraphAny
GraphAny: Fully-inductive Node Classification on Arbitrary Graphs
DeepGraphLearning/RNNLogic
DeepGraphLearning/ESM-GearNet
ESM-GearNet for Protein Structure Representation Learning (https://arxiv.org/abs/2303.06275)
DeepGraphLearning/ProtST
[ICML-23 ORAL] ProtST: Multi-Modality Learning of Protein Sequences and Biomedical Texts
DeepGraphLearning/PEER_Benchmark
PEER Benchmark, appear at NeurIPS 2022 Dataset and Benchmark Track (https://arxiv.org/abs/2206.02096)
DeepGraphLearning/DiffPack
Implementation of DiffPack: A Torsional Diffusion Model for Autoregressive Protein Side-Chain Packing
DeepGraphLearning/GNN-QE
Official implementation of Graph Neural Network Query Executor (ICML 2022)
DeepGraphLearning/GearBind
Pretrainable geometric graph neural network for antibody affinity maturation
DeepGraphLearning/InductiveQE
Official implementation of Inductive Logical Query Answering in Knowledge Graphs (NeurIPS 2022)
DeepGraphLearning/S3F
Sequence-Structure-Surface Model for Protein Fitness Prediction (S3F)
DeepGraphLearning/SiamDiff
Code for Pre-training Protein Encoder via Siamese Sequence-Structure Diffusion Trajectory Prediction (https://arxiv.org/abs/2301.12068)
DeepGraphLearning/esm-s
Structure-Informed Protein Language Model
DeepGraphLearning/scCello
[NeurIPS 2024 Spotlight] A cell-ontology guided transcriptome foundation model
DeepGraphLearning/HAX
Overcoming Long-Context Limitations of State-Space Models via Context-Dependent Sparse Attention (NeurIPS 2025)
DeepGraphLearning/ProteinTutorial_AAAI2025
https://deepgraphlearning.github.io/ProteinTutorial_AAAI2025/
DeepGraphLearning/torchdrug-site
Website for TorchDrug
DeepGraphLearning/ultra_torchdrug
A TorchDrug version of ULTRA for reproducibility
DeepGraphLearning/torchprotein-site
Website for TorchProtein
DeepGraphLearning/Math80600A_2023W
DeepGraphLearning/MATH60629A_2025A
DeepGraphLearning/MATH60630A_2024A
DeepGraphLearning/Math80648A_2025A
DeepGraphLearning/project
Website for research projects