weihua916

taichi

Productive, portable, and performant GPU programming in Python.

CLIP

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

sentence-transformers

Multilingual Sentence & Image Embeddings with BERT

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

alphafold

Open source code for AlphaFold.

gcn

Implementation of Graph Convolutional Networks in TensorFlow

conda

A system-level, binary package and environment manager running on all major operating systems and platforms.

rdkit

The official sources for the RDKit library

cs224u

Code for Stanford CS224u

Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.

ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning

DIG

A library for graph deep learning research

pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

PGL

Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework based on PaddlePaddle

pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

KnowledgeGraphEmbedding

arxiv-style

A Latex style and template for paper preprints (based on NIPS style)

pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

KILT

Library for Knowledge Intensive Language Tasks

PaddleHelix

Bio-Computing Platform Featuring Large-Scale Representation Learning and Multi-Task Deep Learning “螺旋桨”生物计算工具集

pytorch_cluster

PyTorch Extension Library of Optimized Graph Cluster Algorithms

wilds

A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.

LinkBERT

[ACL 2022] LinkBERT: A Knowledgeable Language Model 😎 Pretrained with Document Links

SEAL_OGB

An open-source implementation of SEAL for link prediction in open graph benchmark (OGB) datasets.

pyg_autoscale

Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch

FLAG

Official implementation of our FLAG paper (CVPR2022)

pprgo_pytorch

PPRGo model in PyTorch, as proposed in "Scaling Graph Neural Networks with Approximate PageRank" (KDD 2020)

gridai

Placeholder for the opensource Grid AI components

bc-emb

cs224w