Inference code for LLaMA models

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch tutorials.

Datasets, Transforms and Models specific to Computer Vision

Ahead-of-time compilation of PyTorch JIT for NVIDIA GPUs

An easy to use PyTorch to TensorRT converter

VMZ: Model Zoo for Video Modeling

📖The Big-&-Extending-Repository-of-Transformers: Pretrained PyTorch models for Google's BERT, OpenAI GPT & GPT-2, Google/CMU Transformer-XL.

TVM integration into PyTorch

Open Neural Network Exchange

A mix of GAN implementations including progressive growing

Deep Learning Examples