pytorch

pytorch

Geek Repo

Location:where the eigens are valued

Home Page:https://pytorch.org

Github PK Tool:Github PK Tool

ezoic increase your site revenue

pytorch's repositories

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Language:C++License:NOASSERTIONStargazers:56299Issues:1595Issues:26253

examples

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

Language:PythonLicense:BSD-3-ClauseStargazers:18446Issues:391Issues:524

vision

Datasets, Transforms and Models specific to Computer Vision

Language:PythonLicense:BSD-3-ClauseStargazers:11702Issues:363Issues:2345

tutorials

PyTorch tutorials.

Language:PythonLicense:BSD-3-ClauseStargazers:6048Issues:161Issues:540

ignite

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

Language:PythonLicense:BSD-3-ClauseStargazers:3970Issues:61Issues:1047

captum

Model interpretability and understanding for PyTorch

Language:PythonLicense:BSD-3-ClauseStargazers:3197Issues:202Issues:355

text

Data loaders and abstractions for text and NLP

Language:PythonLicense:BSD-3-ClauseStargazers:3014Issues:259Issues:650

glow

Compiler for Neural Network hardware accelerators

Language:C++License:Apache-2.0Stargazers:2751Issues:149Issues:796

serve

Serve, optimize and scale PyTorch models in production

Language:JavaLicense:Apache-2.0Stargazers:2620Issues:47Issues:883

botorch

Bayesian optimization in PyTorch

Language:PythonLicense:MITStargazers:2274Issues:48Issues:263

audio

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

Language:PythonLicense:BSD-2-ClauseStargazers:1670Issues:63Issues:596

xla

Enabling PyTorch on Google TPU

Language:C++License:NOASSERTIONStargazers:1631Issues:42Issues:1044

TensorRT

PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

Language:Jupyter NotebookLicense:BSD-3-ClauseStargazers:1433Issues:43Issues:518

opacus

Training PyTorch models with differential privacy

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:1149Issues:41Issues:176

torchrec

Pytorch domain library for recommendation systems

Language:PythonLicense:BSD-3-ClauseStargazers:1000Issues:20Issues:29

functorch

functorch is JAX-like composable function transforms for PyTorch.

Language:PythonLicense:BSD-3-ClauseStargazers:959Issues:21Issues:370

FBGEMM

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

Language:C++License:NOASSERTIONStargazers:813Issues:34Issues:64

data

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

Language:PythonLicense:BSD-3-ClauseStargazers:702Issues:21Issues:153

benchmark

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

Language:PythonLicense:BSD-3-ClauseStargazers:376Issues:180Issues:248

live

PyTorch Live is an easy to use library of tools for creating on-device ML demos on Android and iOS. With Live, you can build a working mobile ML prototype in minutes.

Language:TypeScriptLicense:MITStargazers:301Issues:19Issues:49

kineto

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

Language:HTMLLicense:NOASSERTIONStargazers:277Issues:14Issues:114

nestedtensor

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

Language:Jupyter NotebookLicense:BSD-3-ClauseStargazers:230Issues:20Issues:67

pytorch.github.io

The website for PyTorch

Language:Jupyter NotebookLicense:BSD-3-ClauseStargazers:156Issues:20Issues:165

torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

Language:PythonLicense:BSD-3-ClauseStargazers:155Issues:22Issues:120

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:BSD-3-ClauseStargazers:140Issues:12Issues:92

cppdocs

PyTorch C++ API Documentation

csprng

Cryptographically secure pseudorandom number generators for PyTorch

Language:BatchfileLicense:BSD-3-ClauseStargazers:77Issues:20Issues:43

PiPPy

Pipeline Parallelism for PyTorch

Language:PythonLicense:BSD-3-ClauseStargazers:38Issues:15Issues:82

maskedtensor

MaskedTensors for PyTorch

Language:PythonLicense:NOASSERTIONStargazers:22Issues:2Issues:5

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:15Issues:6Issues:66