Explore @pytorch Open Source projects

Top Rating
- Top Contributors
  Discover the Top Open Source contributors by country or by language
- Interviews
  Discover real stories from Open Source developers
Discover

Discover your Favorite Language
Discover the top trending repositories and projects on Github. Explore the latest trends in your preferred languages.

C#

TypeScript

PHP

R

OCaml

Lua

Emacs Lisp

C

More Languages
Awesome

Awesome repositories
Discover the most awesome repositories and projects of your favorite languages. Inspired by the Awesome-* lists trend in GitHub.

Go

C#

Scala

Rust

Dart

C

Ada

PHP

More Languages
By Country

Rankings by Country
Discover the community of talented open source contributors in each country.

🇱🇸 Lesotho

🇧🇦 Bosnia and Herzegovina

🇵🇼 Palau

🇦🇽 Åland Islands

🇹🇰 Tokelau

🇹🇷 Turkey

🇮🇶 Iraq

🇧🇲 Bermuda

All Countries Compare Countries

@pytorch

pytorch

Stars
192,451
Global Org. Rank 47 (Top 0.02 %)
Registered almost 8 years ago
Most used languages

Python
48.6 %

C++
11.4 %

Jupyter Notebook
10.0 %

HTML
5.7 %

TypeScript
4.3 %

C
4.3 %

JavaScript
2.9 %

Java
2.9 %

Swift
1.4 %

Groovy
1.4 %

Makefile 1.4 %

Haskell
1.4 %

Shell
1.4 %

Batchfile 1.4 %

CSS
1.4 %

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

examples

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

vision

Datasets, Transforms and Models specific to Computer Vision

tutorials

PyTorch tutorials.

Jupyter Notebook

captum

Model interpretability and understanding for PyTorch

ignite

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

serve

Serve, optimize and scale PyTorch models in production

torchtune

A Native-PyTorch Library for LLM Fine-tuning

text

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

ELF

ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation

glow

Compiler for Neural Network hardware accelerators

botorch

Bayesian optimization in PyTorch

Jupyter Notebook

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

audio

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

xla

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

rl

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

torchrec

Pytorch domain library for recommendation systems

opacus

Training PyTorch models with differential privacy

Jupyter Notebook

tnt

A lightweight library for PyTorch training tools and utilities

QNNPACK

Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators

executorch

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

android-demo-app

PyTorch android examples of usage in applications

functorch

functorch is JAX-like composable function transforms for PyTorch.

Jupyter Notebook

hub

Submission to https://pytorch.org/hub/

torchtitan

A native PyTorch Library for large model training

FBGEMM

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

data

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

torchdynamo

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

cpuinfo

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

extension-cpp

C++ extensions in PyTorch

translate

Translate - a PyTorch Language Library

benchmark

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

elastic

PyTorch elastic training

PiPPy

Pipeline Parallelism for PyTorch

kineto

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

torcharrow

High performance model preprocessing library on PyTorch

tensordict

TensorDict is a pytorch dedicated tensor container.

ios-demo-app

PyTorch iOS examples

tvm

TVM integration into PyTorch

contrib

Implementations of ideas from recent papers

ort

Accelerate PyTorch models with ONNX Runtime

builder

Continuous builder and binary build scripts for pytorch

accimage

high performance image loading and augmenting routines mimicking PIL.Image interface

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.

ao

Native PyTorch library for quantization and sparsity

extension-ffi

Examples of C extensions for PyTorch

nestedtensor

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

Jupyter Notebook

tensorpipe

A tensor-aware point-to-point communication primitive for machine learning

pytorch.github.io

The website for PyTorch

hydra-torch

Configuration classes enabling type-safe PyTorch configuration for Hydra apps

workshops

This is a repository for all workshop related materials.

Jupyter Notebook

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.

cppdocs

PyTorch C++ API Documentation

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.

torchsnapshot

A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.

java-demo

Jupyter Notebook

torchdistx

Torch Distributed Experimental

rfcs

PyTorch RFCs (experimental)

extension-script

Example repository for custom C++/CUDA operators for TorchScript

csprng

Cryptographically secure pseudorandom number generators for PyTorch

pytorch_sphinx_theme

PyTorch Sphinx Theme

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.

maskedtensor

MaskedTensors for PyTorch

add-annotations-github-action

A GitHub action to run clang-tidy and annotate failures

ci-hud

HUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressions

probot

PyTorch GitHub bot written in probot

ossci-job-dsl

Jenkins job definitions for OSSCI

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

torchhub_testing

Repo to test torchhub. Nothing to see here.

docs

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

dr-ci

Diagnose and remediate CI jobs

pytorch-ci-dockerfiles

Scripts for generating docker images for PyTorch CI

labeler-github-action

GitHub action for labeling issues and pull requests based on conditions