pytorch

where the eigens are valued

https://pytorch.org

pytorch's repositories

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Language:PythonNOASSERTION82284 1747 45149

vision

Datasets, Transforms and Models specific to Computer Vision

Language:PythonBSD-3-Clause15998 434 3240

serve

Serve, optimize and scale PyTorch models in production

Language:JavaApache-2.04160 57 1625

torchtune

A Native-PyTorch Library for LLM Fine-tuning

Language:PythonBSD-3-Clause3964 45 528

text

Models, data loaders and abstractions for language processing, powered by PyTorch

Language:PythonBSD-3-Clause3495 308 796

torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

Language:PythonBSD-3-Clause3131 37 272

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookMIT3064 51 516

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language:PythonBSD-3-Clause2515 71 1488

audio

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

Language:PythonBSD-2-Clause2486 73 931

xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

Language:C++NOASSERTION2450 57 2166

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

Language:PythonMIT2215 40 606

torchtitan

A native PyTorch Library for large model training

Language:PythonBSD-3-Clause1988 36 134

torchrec

Pytorch domain library for recommendation systems

Language:PythonBSD-3-Clause1874 32 168

executorch

On-device AI across mobile, embedded and edge for PyTorch

Language:C++NOASSERTION1715 54 392

opacus

Training PyTorch models with differential privacy

Language:Jupyter NotebookApache-2.01680 45 292

tnt

A lightweight library for PyTorch training tools and utilities

Language:PythonNOASSERTION1658 45 71

functorch

functorch is JAX-like composable function transforms for PyTorch.

Language:Jupyter NotebookBSD-3-Clause1389 27 522

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Language:C++NOASSERTION1171 66 162

data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

Language:PythonBSD-3-Clause1117 36 489

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

Language:PythonBSD-3-Clause846 242 888

tensordict

TensorDict is a pytorch dedicated tensor container.

Language:PythonMIT809 28 107

ao

PyTorch native quantization and sparsity for training and inference

Language:PythonBSD-3-Clause751 35 197

builder

Continuous builder and binary build scripts for pytorch

Language:ShellBSD-2-Clause327 24 133

torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

Language:PythonNOASSERTION326 21 180

pytorch.github.io

The website for PyTorch

Language:HTMLBSD-3-Clause218 23 245

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.

Language:PythonNOASSERTION214 15 32

cppdocs

PyTorch C++ API Documentation

Language:HTML206 27 11

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.

Language:TypeScriptNOASSERTION77 15 2978

torchcodec

PyTorch video decoding

Language:PythonBSD-3-Clause48 4 26

docs

This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.

Language:HTMLBSD-3-Clause4 5 2