blzheng's repositories
DeepSpeedSYCLSupport
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch-YOLOv4
PyTorch ,ONNX and TensorRT implementation of YOLOv4
RobustVideoMatting
Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
torchbenchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
torchdynamo
A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
transformers
🤗 Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.
TransNetV2
TransNet V2: Shot Boundary Detection Neural Network
USRNet
Deep Unfolding Network for Image Super-Resolution (CVPR, 2020) (PyTorch)
vision
Datasets, Transforms and Models specific to Computer Vision