pytorch

pytorch

Geek Repo

Location:where the eigens are valued

Home Page:https://pytorch.org

Github PK Tool:Github PK Tool

pytorch's repositories

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Language:PythonLicense:NOASSERTIONStargazers:77025Issues:1695Issues:40805

vision

Datasets, Transforms and Models specific to Computer Vision

Language:PythonLicense:BSD-3-ClauseStargazers:15287Issues:406Issues:3127

tutorials

PyTorch tutorials.

Language:Jupyter NotebookLicense:BSD-3-ClauseStargazers:7740Issues:178Issues:724

ignite

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

Language:PythonLicense:BSD-3-ClauseStargazers:4441Issues:62Issues:1366

serve

Serve, optimize and scale PyTorch models in production

Language:JavaLicense:Apache-2.0Stargazers:3929Issues:58Issues:1547

text

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

Language:PythonLicense:BSD-3-ClauseStargazers:3430Issues:283Issues:787

glow

Compiler for Neural Network hardware accelerators

Language:C++License:Apache-2.0Stargazers:3122Issues:156Issues:826

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookLicense:MITStargazers:2928Issues:56Issues:461

audio

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

Language:PythonLicense:BSD-2-ClauseStargazers:2368Issues:75Issues:899

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language:PythonLicense:BSD-3-ClauseStargazers:2297Issues:70Issues:1344

xla

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

Language:C++License:NOASSERTIONStargazers:2259Issues:58Issues:1737

rl

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

Language:PythonLicense:MITStargazers:1805Issues:42Issues:495

torchrec

Pytorch domain library for recommendation systems

Language:PythonLicense:BSD-3-ClauseStargazers:1701Issues:27Issues:130

tnt

A lightweight library for PyTorch training tools and utilities

Language:PythonLicense:NOASSERTIONStargazers:1612Issues:47Issues:68

functorch

functorch is JAX-like composable function transforms for PyTorch.

Language:Jupyter NotebookLicense:BSD-3-ClauseStargazers:1367Issues:27Issues:519

data

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

Language:PythonLicense:BSD-3-ClauseStargazers:1062Issues:36Issues:471

FBGEMM

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

Language:C++License:NOASSERTIONStargazers:1054Issues:52Issues:135

cpuinfo

CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

Language:CLicense:BSD-2-ClauseStargazers:917Issues:233Issues:76

benchmark

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

Language:PythonLicense:BSD-3-ClauseStargazers:764Issues:218Issues:820

torcharrow

High performance model preprocessing library on PyTorch

Language:PythonLicense:BSD-3-ClauseStargazers:626Issues:24Issues:61

kineto

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

Language:HTMLLicense:NOASSERTIONStargazers:585Issues:23Issues:186

PiPPy

Pipeline Parallelism for PyTorch

Language:PythonLicense:BSD-3-ClauseStargazers:555Issues:34Issues:203

builder

Continuous builder and binary build scripts for pytorch

Language:ShellLicense:BSD-2-ClauseStargazers:318Issues:27Issues:109

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:PythonLicense:NOASSERTIONStargazers:286Issues:17Issues:173

pytorch.github.io

The website for PyTorch

Language:HTMLLicense:BSD-3-ClauseStargazers:209Issues:25Issues:235

cppdocs

PyTorch C++ API Documentation

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:PythonLicense:NOASSERTIONStargazers:180Issues:16Issues:25

rfcs

PyTorch RFCs (experimental)

pytorch_sphinx_theme

PyTorch Sphinx Theme

Language:CSSLicense:MITStargazers:91Issues:11Issues:41

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:TypeScriptLicense:NOASSERTIONStargazers:63Issues:15Issues:2851