wizzniu

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

pytorch

Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch

distributions

NodeSource Node.js Binary Distributions

HolisticTraceAnalysis

A library to analyze PyTorch traces.

kineto

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

training

Reference implementations of MLPerf™ training benchmarks

SKNet

Code for our CVPR 2019 paper: Selective Kernel Networks; See zhihu：https://zhuanlan.zhihu.com/p/59690223

SENet

Squeeze-and-Excitation Networks

AttentionConvLSTM

"Attention in Convolutional LSTM for Gesture Recognition" in NIPS 2018

Non-local_pytorch

Implementation of Non-local Block.

video-nonlocal-net

Non-local Neural Networks for Video Classification

ConvLSTM-PyTorch

ConvLSTM/ConvGRU (Encoder-Decoder) with PyTorch on Moving-MNIST

ConvLSTM_pytorch

Implementation of Convolutional LSTM in PyTorch.

MDL

a pytorch version implement of paper "Flow Prediction in Spatio-Temporal Networks Based on Multitask Deep Learning"

STResNet-PyTorch

A Pytorch Implementation of STResNet, AAAI 2017

UrbanFM

Pytorch Implementation of Urban Flow Magnifier (UrbanFM), KDD-19

ml-visuals

🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.

PGL

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

PaddleSpatial

PaddleSpatial is an open-source spatial-temporal computing tool based on PaddlePaddle.

Bigscity-LibCity-PaperList

Paper list in traffic prediction field

Bigscity-LibCity

LibCity: An Open Library for Urban Spatial-temporal Data Mining

DL-Traff-Graph

[CIKM 2021 Resource Paper] DL-Traff: Survey and Benchmark of Deep Learning Models for Urban Traffic Prediction (Graph Part)

DL-Traff-Grid

[CIKM 2021 Resource Paper] DL-Traff: Survey and Benchmark of Deep Learning Models for Urban Traffic Prediction (Grid Part)

paper-reading

深度学习经典、新论文逐段精读

data-mining-conferences

Ranking, acceptance rate, deadline, and publication tips

Conference-Acceptance-Rate

Acceptance rates for the major AI conferences

xgboost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

google-research

Google Research

tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf

ensemble-learning