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:PythonNOASSERTION80639 1738 43430

vision

Datasets, Transforms and Models specific to Computer Vision

Language:PythonBSD-3-Clause15783 427 3191

tutorials

PyTorch tutorials.

Language:Jupyter NotebookBSD-3-Clause7996 177 772

serve

Serve, optimize and scale PyTorch models in production

Language:JavaApache-2.04071 59 1612

torchtune

A Native-PyTorch Library for LLM Fine-tuning

Language:PythonBSD-3-Clause3636 43 374

text

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

Language:PythonBSD-3-Clause3480 301 791

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookMIT3017 53 498

audio

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

Language:PythonBSD-2-Clause2442 73 921

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language:PythonBSD-3-Clause2428 69 1445

xla

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

Language:C++NOASSERTION2386 55 2089

rl

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

Language:PythonMIT2032 41 564

torchrec

Pytorch domain library for recommendation systems

Language:PythonBSD-3-Clause1805 29 151

opacus

Training PyTorch models with differential privacy

Language:Jupyter NotebookApache-2.01655 43 286

tnt

A lightweight library for PyTorch training tools and utilities

Language:PythonNOASSERTION1643 45 70

executorch

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

Language:C++NOASSERTION1458 54 315

functorch

functorch is JAX-like composable function transforms for PyTorch.

Language:Jupyter NotebookBSD-3-Clause1380 28 520

torchtitan

A native PyTorch Library for large model training

Language:PythonBSD-3-Clause1363 36 107

FBGEMM

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

Language:C++NOASSERTION1116 55 146

data

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

Language:PythonBSD-3-Clause1088 37 484

benchmark

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

Language:PythonBSD-3-Clause824 232 866

tensordict

TensorDict is a pytorch dedicated tensor container.

Language:PythonMIT678 31 101

kineto

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

Language:HTMLNOASSERTION661 27 207

ao

Custom data types and layouts for training and inference

Language:PythonBSD-3-Clause428 26 88

builder

Continuous builder and binary build scripts for pytorch

Language:ShellBSD-2-Clause323 25 128

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:PythonNOASSERTION311 21 179

pytorch.github.io

The website for PyTorch

Language:HTMLBSD-3-Clause215 25 242

cppdocs

PyTorch C++ API Documentation

Language:HTML201 28 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:TypeScriptNOASSERTION72 16 2958

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

Language:MakefileMIT5 150

docs

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

Language:HTMLBSD-3-Clause4 6 1