hongsunjang

guided-diffusion

GNN-ARCH

[ASAP 2020; FPGA 2020] Hardware architecture to accelerate GNNs (common IP modules for minibatch training and full batch inference)

Cosmos-plus-OpenSSD

Cosmos OpenSSD + Hardware and Software source distribution

Large_Scale_GCN_Benchmarking

This is an authors' implementation of the NIPS 2022 dataset and Benchmark Track Paper "A Comprehensive Study on Large Scale Graph Training: Benchmarking and Rethinking" in PyTorch.

pytorch-diffusion

A basic PyTorch implementation of 'Denoising Diffusion Probabilistic Models'

Neural-Networks-on-Silicon

This is originally a collection of papers on neural network accelerators. Now it's more like my selection of research on deep learning and computer architecture.

accelergy

Accelergy is an energy estimation infrastructure for accelerator energy estimations

timeloop

Timeloop performs modeling, mapping and code-generation for tensor algebra workloads on various accelerator architectures.

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Optimus-CC

[ASPLOS'23] Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression

llcg

Source code for Learn Locally Correct Globally

awesome-gnn-systems

A list of awesome GNN systems.

One-sentence_Diffusion_summary

The repo for studying and sharing diffusion models.

Vundle.vim

Vundle, the plug-in manager for Vim

vim-autoformat

Provide easy code formatting in Vim by integrating existing code formatters.

NeutronStarLite

A Distributed GNN system

nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

task-spooler

A scheduler for GPU/CPU tasks

DeeperGCN-dgl

A DGL implementation of "DeeperGCN: All You Need to Train Deeper GCNs".

PipeGCN

[ICLR 2022] "PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication" by Cheng Wan, Youjie Li, Cameron R. Wolfe, Anastasios Kyrillidis, Nam Sung Kim, Yingyan Lin

pytorch_geometric

Graph Neural Network Library for PyTorch

PaGraph

SoCC'20 and TPDS'21: Scaling GNN Training on Large Graphs via Computation-aware Caching and Partitioning.

deep_gcns_torch

Pytorch Repo for DeepGCNs (ICCV'2019 Oral, TPAMI'2021), DeeperGCN (arXiv'2020) and GNN1000(ICML'2021): https://www.deepgcns.org

lightly

A python library for self-supervised learning on images.

dgl

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

torch-quiver

PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.

simsiam

PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566

barlowtwins

PyTorch implementation of Barlow Twins.

SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations