cpuhrsch Goto Github PK

audio

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

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

detr

End-to-End Object Detection with Transformers

examples

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

fairseq

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

fairseq2

FAIR Sequence Modeling Toolkit 2

maskedtensor

MaskedTensors for PyTorch

mingpt

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

nestedtensor

parlai

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

protoquant

Prototype routines for GPU quantization written using PyTorch.

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch.github.io

The website for PyTorch

reproducentissue

rfcs

PyTorch RFCs (experimental)

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.

sleef

SIMD library for evaluating elementary functions, a vectorized libm

text

Data loaders and abstractions for text and NLP

torchcsr

Playing with csr layouts

torchfused

transformers

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

tutorials

PyTorch tutorials.

vision

Datasets, Transforms and Models specific to Computer Vision

xla

Enabling PyTorch on Google TPU