jundu's repositories
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
3D-GAN
A simple and unofficial 3D-GAN implementation using PyTorch
BasicVSR
Official repository of "BasicVSR++: Improving Video Super-Resolution with Enhanced Propagation and Alignment"
beit-large
generate from: https://huggingface.co/microsoft/beit-large-patch16-224-pt22k-ft22k
Cascade_R-CNN_ResNet101-FPN
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
conformer
PyTorch implementation of "Conformer: Convolution-augmented Transformer for Speech Recognition" (INTERSPEECH 2020)
ConvNeXt
Code release for ConvNeXt model
DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
deit
Official DeiT repository
DRN-D-54
Dilated Residual Networks
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
F3Net
Codes for the AAAI 2020 paper "F3Net: Fusion, Feedback and Focus for Salient Object Detection"
fairseq
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
fastformers
FastFormers - highly efficient transformer models for NLU
FastSpeech2
An implementation of Microsoft's "FastSpeech 2: Fast and High-Quality End-to-End Text to Speech"
intel-extension-for-deepspeed
Intel® Extension for DeepSpeed* is an extension that brings Intel GPU (XPU) support to DeepSpeed.
MalConv
Pytorch implementation of MalConv
mlperf_training
Reference implementations of MLPerf™ training benchmarks
pretrained-models.pytorch
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch-3dunet
3D U-Net model for volumetric semantic segmentation written in pytorch
pytorch-dnc
Differentiable Neural Computers, Sparse Access Memory and Sparse Differentiable Neural Computers, for Pytorch
PyTorch-GAN
PyTorch implementations of Generative Adversarial Networks.
pytorch-ssd
MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
SlowFast
PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
Swin-Transformer
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
Visualbert
refer to https://colab.research.google.com/drive/1bLGxKdldwqnMVA5x4neY7-l_8fKGWQYI?usp=sharing#scrollTo=9Qv-4b7LYOVw
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Xception
A PyTorch implementation of Xception: Deep Learning with Depthwise Separable Convolutions
yolov5
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite