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:C++NOASSERTION64174 1629 32257

examples

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

Language:PythonBSD-3-Clause19901 399 576

vision

Datasets, Transforms and Models specific to Computer Vision

Language:PythonBSD-3-Clause13537 387 2776

tutorials

PyTorch tutorials.

Language:PythonBSD-3-Clause6794 177 609

ignite

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

Language:PythonBSD-3-Clause4229 60 1179

serve

Serve, optimize and scale PyTorch models in production

Language:JavaApache-2.03254 53 1170

text

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

Language:PythonBSD-3-Clause3242 265 733

glow

Compiler for Neural Network hardware accelerators

Language:C++Apache-2.02930 158 816

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookMIT2557 51 376

audio

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

Language:PythonBSD-2-Clause2027 74 749

xla

Enabling PyTorch on Google TPU

Language:C++NOASSERTION1899 54 1332

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language:Jupyter NotebookBSD-3-Clause1851 63 836

tnt

A lightweight library for PyTorch training tools and utilities

Language:PythonNOASSERTION1510 49 60

torchrec

Pytorch domain library for recommendation systems

Language:PythonBSD-3-Clause1344 27 76

functorch

functorch is JAX-like composable function transforms for PyTorch.

Language:Jupyter NotebookBSD-3-Clause1263 26 517

rl

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

Language:PythonMIT1024 34 238

data

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

Language:PythonBSD-3-Clause929 26 404

FBGEMM

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

Language:C++NOASSERTION907 48 99

torcharrow

High performance model preprocessing library on PyTorch

Language:PythonBSD-3-Clause570 21 55

benchmark

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

Language:PythonBSD-3-Clause553 194 498

kineto

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

Language:HTMLNOASSERTION419 20 146

builder

Continuous builder and binary build scripts for pytorch

Language:ShellBSD-2-Clause254 26 84

tau

Pipeline Parallelism for PyTorch

Language:PythonBSD-3-Clause235 32 166

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:PythonBSD-3-Clause205 17 130

pytorch.github.io

The website for PyTorch

Language:HTMLBSD-3-Clause182 21 202

cppdocs

PyTorch C++ API Documentation

Language:HTML150 21 11

multipy

torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.

Language:C++NOASSERTION113 12 55

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:PythonNOASSERTION105 15 6

csprng

Cryptographically secure pseudorandom number generators for PyTorch

Language:BatchfileBSD-3-Clause84 29 43

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:TypeScriptNOASSERTION32 15 2619