🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Show, Attend, and Tell | a PyTorch Tutorial to Image Captioning

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch