so-dium Goto Github PK

996.icu

Repo for counting stars and contributing. Press F to pay respect to glorious developers.

balanced-dataparallel

这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量

biochemgnn_dense

blink

Entity Linker solution

coca

CIKM 2021: Contrastive Learning of User Behavior Sequence for Context-Aware Document Ranking

codetemplate

colake

CoLAKE: Contextualized Language and Knowledge Embedding

constraintgraph4nso

AAAI 2021: Neural Sentence Ordering Based on Constraint Graphs

conv-knrm

Convolutional Neural Networks for Soft-Matching N-Grams in Ad-hoc Search

graphormer

Graphormer is a deep learning package that allows researchers and developers to train custom models for molecule modeling tasks. It aims to accelerate the research and application in AI for molecule science, such as material discovery, drug discovery, etc.

intra-document-cascade

k-nrm

K-NRM: End-to-End Neural Ad-hoc Ranking with Kernel Pooling

knowledge_graph_attention_network

KGAT: Knowledge Graph Attention Network for Recommendation, KDD2019

m2div

matchzoo

Facilitating the design, comparison and sharing of deep text matching models.

nltk

NLTK Source

pssl

pygat

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

pytorch-distributed

A quickstart and benchmark for pytorch distributed training.

pytorch-gat

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

pytorch_distributedparallel_gpu_test

Pytorch distributed data parallel test of GPU on MNIST

researchfigure

Some example codes for drawing figures in research paper

shawn

simcse

EMNLP'2021: SimCSE: Simple Contrastive Learning of Sentence Embeddings

supcontrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

test

text-to-text-transfer-transformer

Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"

transformers

🤗 Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.

trec-deep-learning-quick-start

warc

Python library for reading and writing warc files (Especially for Clueweb/Trec Datasets)