pytorch

Top repositories by stars

• pytorch/pytorch(on leaderboard)

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  Python97,483
• pytorch/examples

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

  Python23,745
• pytorch/vision

  Datasets, Transforms and Models specific to Computer Vision

  Python17,517
• pytorch/tutorials

  PyTorch tutorials.

  Python9,044
• pytorch/torchtitan

  A PyTorch native platform for training generative AI models

  Python5,076
• pytorch/ignite

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

  Python4,735
• pytorch/serve

  Serve, optimize and scale PyTorch models in production

  Java4,359
• pytorch/executorch

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

  Python4,277
• pytorch/torchchat

  Run PyTorch LLMs locally on servers, desktop and mobile

  Python3,624
• pytorch/text

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

  Python3,565
• pytorch/ELF

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

  C++3,415
• pytorch/glow

  Compiler for Neural Network hardware accelerators

  C++3,324
• pytorch/rl

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

  Python3,305
• pytorch/TensorRT

  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

  Python2,947
• pytorch/audio

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

  Python2,831
• pytorch/xla

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

  C++2,749
• pytorch/ao

  PyTorch native quantization and sparsity for training and inference

  Python2,691
• pytorch/android-demo-app

  PyTorch android examples of usage in applications

  Java1,556
• pytorch/QNNPACK

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

  C1,546
• pytorch/FBGEMM

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

  C++1,535
• pytorch/functorch

  functorch is JAX-like composable function transforms for PyTorch.

  Jupyter Notebook1,436
• pytorch/hub

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

  Python1,432
• pytorch/gloo

  Collective communications library with various primitives for multi-machine training.

  C++1,398
• pytorch/extension-cpp

  C++ extensions in PyTorch

  Python1,183
• pytorch/cpuinfo

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

  C1,153
• pytorch/torchdynamo

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

  Python1,074
• pytorch/benchmark

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

  Python1,014
• pytorch/tensordict

  TensorDict is a pytorch dedicated tensor container.

  Python1,008
• pytorch/kineto

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

  HTML922
• pytorch/translate

  Translate - a PyTorch Language Library

  Python833
• pytorch/PiPPy

  Pipeline Parallelism for PyTorch

  Python785
• pytorch/helion

  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

  Python758
• pytorch/elastic

  PyTorch elastic training

  Python728
• pytorch/torcharrow

  High performance model preprocessing library on PyTorch

  Python646
• pytorch/ios-demo-app

  PyTorch iOS examples

  Swift611
• pytorch/tvm

  TVM integration into PyTorch

  C++456
• pytorch/contrib

  Implementations of ideas from recent papers

  Python392
• pytorch/ort

  Accelerate PyTorch models with ONNX Runtime

  Python368
• pytorch/builder

  Continuous builder and binary build scripts for pytorch

  Python356
• pytorch/accimage

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

  C318
• pytorch/tensorpipe

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

  C++284
• pytorch/pytorch.github.io

  The website for PyTorch

  HTML276
• pytorch/extension-ffi

  Examples of C extensions for PyTorch

  Python256
• pytorch/nestedtensor

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

  Jupyter Notebook250
• pytorch/cppdocs

  PyTorch C++ API Documentation

  HTML246
• pytorch/rfcs

  PyTorch RFCs (experimental)

  138
• pytorch/java-demo

  Jupyter Notebook131
• pytorch/torchdistx

  Torch Distributed Experimental

  Python117
• pytorch/extension-script

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

  Python114
• pytorch/test-infra

  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

  TypeScript106
• pytorch/pytorch_sphinx_theme

  PyTorch Sphinx Theme

  CSS105
• pytorch/expecttest

  Python86
• pytorch/maskedtensor

  MaskedTensors for PyTorch

  Python38
• pytorch/pytorch-integration-testing

  Testing downstream libraries using pytorch release candidates

  Makefile15
• pytorch/add-annotations-github-action

  A GitHub action to run clang-tidy and annotate failures

  JavaScript14
• pytorch/ci-infra

  HCL13
• pytorch/ci-hud

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

  JavaScript13
• pytorch/probot

  PyTorch GitHub bot written in probot

  TypeScript12
• pytorch/docs

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

  HTML10
• pytorch/ossci-job-dsl

  Jenkins job definitions for OSSCI

  Groovy10
• pytorch/pytorch-ci-dockerfiles

  Scripts for generating docker images for PyTorch CI

  5
• pytorch/torchhub_testing

  Repo to test torchhub. Nothing to see here.

  4
• pytorch/caffe2.github.io

  The caffe2 website.

  HTML3
• pytorch/alerting-infra

  Alerting system for the pytorch org

  TypeScript2
• pytorch/dr-ci

  Diagnose and remediate CI jobs

  Haskell2
• pytorch/labeler-github-action

  GitHub action for labeling issues and pull requests based on conditions

  TypeScript2
• pytorch/sccache

  sccache is ccache with cloud storage

  Rust1