A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.
APACHE-2.0 License
Simple Keras-inspired Training Loop for Pytorch.
This repository contains demos I made with the Transformers library by HuggingFace.
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Represent...
PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
Official code for the LoG2022 paper -- DIGRAC: Digraph Clustering Based on Flow Imbalance.
PyTorch Extension Library of Optimized Graph Cluster Algorithms
An unofficial styleguide and best practices summary for PyTorch
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reprod...
Train and run Pytorch models on Apache Spark.
NumPy实现类PyTorch的动态计算图和神经网络框架(MLP, CNN, RNN, Transformer)
A powerful and flexible machine learning platform for drug discovery
Official code for the SDM2022 paper -- SSSNET: Semi-Supervised Signed Network Clustering.
[IEEE Access] "Head Network Distillation: Splitting Distilled Deep Neural Networks for Resource-c...
A native PyTorch Library for large model training
A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computat...