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:PythonNOASSERTION78131 1706 41644

examples

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

Language:PythonBSD-3-Clause21759 397 627

vision

Datasets, Transforms and Models specific to Computer Vision

Language:PythonBSD-3-Clause15474 408 3145

tutorials

PyTorch tutorials.

Language:Jupyter NotebookBSD-3-Clause7821 177 730

captum

Model interpretability and understanding for PyTorch

Language:PythonBSD-3-Clause4584 239 516

ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Language:PythonBSD-3-Clause4458 60 1370

serve

Serve, optimize and scale PyTorch models in production

Language:JavaApache-2.03969 58 1571

text

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

Language:PythonBSD-3-Clause3445 284 789

glow

Compiler for Neural Network hardware accelerators

Language:C++Apache-2.03156 155 827

torchtune

A Native-PyTorch Library for LLM Fine-tuning

Language:PythonBSD-3-Clause3006 37 244

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookMIT2958 53 476

audio

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

Language:PythonBSD-2-Clause2386 73 905

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language:PythonBSD-3-Clause2343 70 1381

xla

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

Language:C++NOASSERTION2300 58 1762

torchrec

Pytorch domain library for recommendation systems

Language:PythonBSD-3-Clause1731 26 139

tnt

A lightweight library for PyTorch training tools and utilities

Language:PythonNOASSERTION1628 46 68

functorch

functorch is JAX-like composable function transforms for PyTorch.

Language:Jupyter NotebookBSD-3-Clause1372 28 520

executorch

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

Language:C++NOASSERTION1071 52 203

FBGEMM

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

Language:C++NOASSERTION1071 52 137

data

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

Language:PythonBSD-3-Clause1070 36 474

torchtitan

A native PyTorch Library for large model training

Language:PythonBSD-3-Clause1068 26 70

benchmark

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

Language:PythonBSD-3-Clause783 220 839

PiPPy

Pipeline Parallelism for PyTorch

Language:PythonBSD-3-Clause623 36 243

kineto

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

Language:HTMLNOASSERTION615 25 199

builder

Continuous builder and binary build scripts for pytorch

Language:ShellBSD-2-Clause319 25 114

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:PythonNOASSERTION299 19 178

ao

Native PyTorch library for quantization and sparsity

Language:PythonBSD-3-Clause212 15 26

pytorch.github.io

The website for PyTorch

Language:HTMLBSD-3-Clause211 24 237

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:TypeScriptNOASSERTION67 14 2872

docs

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

Language:HTMLBSD-3-Clause300