The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Tacotron 2 - PyTorch implementation with faster-than-realtime inference

PyTorch native quantization and sparsity for training and inference

Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.

PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)

A PyTorch-native inference engine with cache acceleration, parallelism and quantization for DiTs.

Fast and accurate face landmark detection library using PyTorch; Support 68-point semi-frontal and 39-point profile landmark detection; Support both coordinate-based and heatmap-based inference; Up to 100 FPS landmark inference speed with SOTA face detector on CPU.

PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference

Neural relational inference for interacting systems - pytorch

🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.

v objective diffusion inference code for PyTorch.

Deep Learning on Flink aims to integrate Flink and deep learning frameworks (e.g. TensorFlow, PyTorch, etc) to enable distributed deep learning training and inference on a Flink cluster.

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.

Automatically split your PyTorch models on multiple GPUs for training & inference

Human segmentation models, training/inference code, and trained weights, implemented in PyTorch

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

Convert pointpillars Pytorch Model To ONNX for TensorRT Inference

Implementation of the ESIM model for natural language inference with PyTorch

Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch

Tacotron 2 - PyTorch implementation with faster-than-realtime inference modified to enable cross lingual voice cloning.

Extensive tutorials for learning how to build deep learning models for causal inference (HTE) using selection on observables in Tensorflow 2 and Pytorch.

PyTorch C++ inference with LibTorch

PyTorch inference for "Progressive Growing of GANs" with CelebA snapshot

TPU inference for vLLM, with unified JAX and PyTorch support.

💎An easy-to-use PyTorch library for face landmarks detection: training, evaluation, inference, and 100+ data augmentations.🎉

BiSeNet: Bilateral Segmentation Network for Real-time Semantic Segmentation | Face Parsing | In PyTorch >> ONNX Runtime Inference

A High-Performance Pytorch Implementation of face detection models, including RetinaFace and DSFD

Serving PyTorch 1.0 Models as a Web Server in C++