Stars
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Python sample codes and textbook for robotics algorithms.
deep learning for image processing including classification and object-detection etc.
OpenAI Baselines: high-quality implementations of reinforcement learning algorithms
Fast and memory-efficient exact attention
pix2tex: Using a ViT to convert images of equations into LaTeX code.
Ongoing research training transformer models at scale
Common used path planning algorithms with animations.
Training and serving large-scale neural networks with auto parallelization.
MOT using deepsort and yolov3 with pytorch
Clean, accessible reproduction of DeepSeek R1-Zero
Multi-Object Tracking via DeepSORT
这是一个faster-rcnn的pytorch实现的库,可以利用voc数据集格式的数据进行训练。
Some Conferences' accepted paper lists (including AI, ML, Robotic)
Graph Transformer Architecture. Source code for "A Generalization of Transformer Networks to Graphs", DLG-AAAI'21.
Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018
Universal Graph Transformer Self-Attention Networks (TheWebConf WWW 2022) (Pytorch and Tensorflow)
[ICRA19] Crowd-aware Robot Navigation with Attention-based Deep Reinforcement Learning
realtime multiple people tracking (centerNet based person detector + deep sort algorithm with pytorch)
Spectral Temporal Graph Neural Network (StemGNN in short) for Multivariate Time-series Forecasting
Human Trajectory Prediction Dataset Benchmark (ACCV 2020)
[ECCV 2020] Code for "Spatio-Temporal Graph Transformer Networks for Pedestrian Trajectory Prediction"
Efficient 6-DoF Grasp Generation in Cluttered Scenes
A Graph Neural Network project on HIV data
[ICCV 2021] Official PyTorch Implementation of "AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting".