Projects
vision
Datasets, Transforms and Models specific to Computer Vision
Python - Released: 09 Nov 2016 - 15,542
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python - Released: 24 Aug 2016 - 22,302
text
Models, data loaders and abstractions for language processing, powered by PyTorch
Python - Released: 12 Dec 2016 - 3,460
audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
Python - Released: 05 May 2017 - 2,468
tnt
A lightweight library for PyTorch training tools and utilities
Python - Released: 10 Dec 2016 - 1,659
torchchat
Run PyTorch LLMs locally on servers, desktop and mobile
Python - Released: 22 Mar 2024 - 3,316
kineto
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML - Released: 16 Jul 2020 - 625
rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Python - Released: 01 Feb 2022 - 1,823
torcheval
A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.
Python - Released: 24 May 2022 - 214
builder
Continuous builder and binary build scripts for pytorch
Python - Released: 02 Sep 2016 - 330
benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Python - Released: 26 May 2017 - 776
test-infra
This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
TypeScript - Released: 30 Mar 2021 - 64