mrshenli's repositories
dummy_collectives
A minimum demo for PyTorch c10d extension APIs
AdaBound
An optimizer that trains as fast as Adam and as good as SGD.
Apache-2.0000
benchmark-1
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
NOASSERTION000
mrshenli.github.io
Github Pages template for academic personal websites, forked from mmistakes/minimal-mistakes
pipeline_experiments
PyTorch pipeline parallelism experiments
pytorchviz
A small package to create visualizations of PyTorch execution graphs
reinforcement-learning
Personal experiments on Reinforcement Learning
tau
Pipeline Parallelism for PyTorch
Language:PythonBSD-3-Clause000