Skip to content

@pytorch pytorch

Change the repository type filter

All

    Repositories list

    67 repositories

    • pytorch

      Public
      Tensors and Dynamic neural networks in Python with strong GPU acceleration
      pythonmachine-learningdeep-learningneural-networkgpunumpyautogradtensor
      Python
      •
      Other
      •26k•96k•16k•1.7k•Updated Dec 16, 2025Dec 16, 2025

    • executorch

      Public
      On-device AI across mobile, embedded and edge for PyTorch
      machine-learningmobileembeddeddeep-learningneural-networkgputensor
      Python
      •
      Other
      •758•3.7k•817•693•Updated Dec 16, 2025Dec 16, 2025

    • tensordict

      Public
      TensorDict is a pytorch dedicated tensor container.
      Python
      •
      MIT License
      •110•992•51•38•Updated Dec 16, 2025Dec 16, 2025

    • rl

      Public
      A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
      machine-learningcontrolreinforcement-learningairoboticsdecision-makingdistributed-computingtorchpytorchrl
      Python
      •
      MIT License
      •424•3.2k•213•95•Updated Dec 16, 2025Dec 16, 2025

    • ort

      Public
      Accelerate PyTorch models with ONNX Runtime
      Python
      •
      MIT License
      •54•368•13•8•Updated Dec 16, 2025Dec 16, 2025

    • kineto

      Public
      A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
      HTML
      •
      Other
      •215•904•63•50•Updated Dec 16, 2025Dec 16, 2025

    • ao

      Public
      PyTorch native quantization and sparsity for training and inference
      trainingsparsitycudainferenceoptimizerpytorchtransformeroffloadingllamaquantization
      Python
      •
      Other
      •386•2.6k•352•198•Updated Dec 16, 2025Dec 16, 2025

    • test-infra

      Public
      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.
      TypeScript
      •
      Other
      •109•103•232•251•Updated Dec 16, 2025Dec 16, 2025

    • pytorch-integration-testing

      Public
      Testing downstream libraries using pytorch release candidates
      Makefile
      •
      Other
      •22•14•1•8•Updated Dec 16, 2025Dec 16, 2025

    • torchtitan

      Public
      A PyTorch native platform for training generative AI models
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •641•4.8k•158•159•Updated Dec 16, 2025Dec 16, 2025

    • FBGEMM

      Public
      FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
      C++
      •
      Other
      •696•1.5k•60•597•Updated Dec 16, 2025Dec 16, 2025

    • TensorRT

      Public
      PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
      machine-learningdeep-learningcudapytorchnvidiajetsontensorrtlibtorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •376•2.9k•268•32•Updated Dec 16, 2025Dec 16, 2025

    • vision

      Public
      Datasets, Transforms and Models specific to Computer Vision
      machine-learningcomputer-vision
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •7.2k•17k•911•270•Updated Dec 16, 2025Dec 16, 2025

    • audio

      Public
      Data manipulation and transformation for audio signal processing, powered by PyTorch
      audiopythonmachine-learningspeechpytorchioaudio-processing
      Python
      •
      BSD 2-Clause "Simplified" License
      •750•2.8k•237•78•Updated Dec 16, 2025Dec 16, 2025

    • ignite

      Public
      High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
      pythonmachine-learningdeep-learningneural-networkmetricspytorchhacktoberfestclosember
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •657•4.7k•112•40•Updated Dec 16, 2025Dec 16, 2025

    • cppdocs

      Public
      PyTorch C++ API Documentation
      HTML
      •36•240•7•3•Updated Dec 16, 2025Dec 16, 2025

    • tutorials

      Public
      PyTorch tutorials.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •4.3k•8.9k•187•41•Updated Dec 16, 2025Dec 16, 2025

    • docs

      Public
      This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.
      HTML
      •
      BSD 3-Clause "New" or "Revised" License
      •7•9•3•0•Updated Dec 16, 2025Dec 16, 2025

    • helion

      Public
      A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •88•682•72•35•Updated Dec 16, 2025Dec 16, 2025

    • benchmark

      Public
      TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
      benchmarkpytorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •329•1k•81•74•Updated Dec 15, 2025Dec 15, 2025

    • cpuinfo

      Public
      CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
      cpucpuidcpu-cachecpu-modelinstruction-setcpu-topology
      C
      •
      BSD 2-Clause "Simplified" License
      •379•1.1k•47•48•Updated Dec 11, 2025Dec 11, 2025

    • pytorch.github.io

      Public
      The website for PyTorch
      HTML
      •
      BSD 3-Clause "New" or "Revised" License
      •312•270•150•63•Updated Dec 11, 2025Dec 11, 2025

    • xla

      Public
      Enabling PyTorch on XLA Devices (e.g. Google TPU)
      deep-learningcompilerpytorchxla
      C++
      •
      Other
      •562•2.7k•694•186•Updated Dec 10, 2025Dec 10, 2025

    • gloo

      Public
      Collective communications library with various primitives for multi-machine training.
      pytorchdistributed-trainingcollectives
      C++
      •
      Other
      •339•1.4k•64•44•Updated Dec 2, 2025Dec 2, 2025

    • ci-infra

      Public
      HCL
      •
      Other
      •5•14•28•2•Updated Nov 26, 2025Nov 26, 2025

    • tensorpipe

      Public
      A tensor-aware point-to-point communication primitive for machine learning
      C++
      •
      Other
      •78•279•47•14•Updated Nov 5, 2025Nov 5, 2025

    • pytorch_sphinx_theme

      Public
      PyTorch Sphinx Theme
      CSS
      •
      MIT License
      •144•104•21•8•Updated Nov 3, 2025Nov 3, 2025

    • alerting-infra

      Public
      Alerting system for the pytorch org
      TypeScript
      •
      BSD 3-Clause "New" or "Revised" License
      •1•2•79•0•Updated Oct 6, 2025Oct 6, 2025

    • torchchat

      Public archive
      Run PyTorch LLMs locally on servers, desktop and mobile
      localpytorchllm
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •246•3.6k•79•15•Updated Sep 10, 2025Sep 10, 2025

    • text

      Public archive
      Models, data loaders and abstractions for language processing, powered by PyTorch
      nlpdeep-learningmodelspytorchdatasetdata-loader
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •812•3.6k•290•56•Updated Sep 10, 2025Sep 10, 2025