cpuhrsch

audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

detr

End-to-End Object Detection with Transformers

maskedtensor

MaskedTensors for PyTorch

nestedtensor

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

reproducentissue

text

Data loaders and abstractions for text and NLP

torchfused

benchmark

blocksparse

Efficient GPU kernels for block-sparse matrix multiplication and convolution

CS344_2021

Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to build in 2021

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

minGPT

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

ParlAI

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.

pytorch.github.io

The website for PyTorch

rfcs

PyTorch RFCs (experimental)

sleef

SIMD library for evaluating elementary functions, a vectorized libm

torchcsr

Playing with csr layouts

transformers

🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.

vision

Datasets, Transforms and Models specific to Computer Vision

xla

Enabling PyTorch on Google TPU

fairseq2

FAIR Sequence Modeling Toolkit 2

protoquant

Prototype routines for GPU quantization written using PyTorch.

seamless_communication

Foundational Models for State-of-the-Art Speech and Text Translation

segment-anything

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.