Wenting Shen's repositories
graph-learn
graph-learn
flash-attention
Fast and memory-efficient exact attention
Language:PythonBSD-3-Clause000
FlashModels
Fast and easy distributed model training examples.
Language:PythonApache-2.0000
graphlearn-for-pytorch
A graph learning library for PyTorch that makes distributed GNN training and inference easy and efficient.
Language:PythonApache-2.0000
LightGBM
A fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks. It is under the umbrella of the DMTK(http://github.com/microsoft/dmtk) project of Microsoft.
pytorch_geometric
Graph Neural Network Library for PyTorch
Language:PythonMIT000
torchdistx
Torch Distributed Experimental
Language:PythonBSD-3-Clause000