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
      •27k•98k•16k•1.9k•Updated Feb 22, 2026Feb 22, 2026

    • 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
      •107•764•82•45•Updated Feb 22, 2026Feb 22, 2026

    • tensordict

      Public
      TensorDict is a pytorch dedicated tensor container.
      Python
      •
      MIT License
      •112•1k•52•46•Updated Feb 22, 2026Feb 22, 2026

    • rl

      Public
      A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
      machine-learningcontrolreinforcement-learningairoboticsdecision-makingdistributed-computingtorchpytorchrl
      Python
      •
      MIT License
      •438•3.3k•203•78•Updated Feb 22, 2026Feb 22, 2026

    • executorch

      Public
      On-device AI across mobile, embedded and edge for PyTorch
      machine-learningmobileembeddeddeep-learningneural-networkgputensor
      Python
      •
      Other
      •848•4.3k•834•787•Updated Feb 22, 2026Feb 22, 2026

    • torchtitan

      Public
      A PyTorch native platform for training generative AI models
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •715•5.1k•182•217•Updated Feb 22, 2026Feb 22, 2026

    • cppdocs

      Public
      PyTorch C++ API Documentation
      HTML
      •38•246•7•3•Updated Feb 22, 2026Feb 22, 2026

    • 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…
      TypeScript
      •
      Other
      •112•106•237•256•Updated Feb 22, 2026Feb 22, 2026

    • ao

      Public
      PyTorch native quantization and sparsity for training and inference
      trainingsparsitycudainferencepytorchtransformerllamaquantizationmxbrrr
      Python
      •
      Other
      •432•2.7k•387•247•Updated Feb 22, 2026Feb 22, 2026

    • 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
      •9•10•3•0•Updated Feb 22, 2026Feb 22, 2026

    • TensorRT

      Public
      PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
      machine-learningdeep-learningcudapytorchnvidiajetsontensorrtlibtorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •382•2.9k•296•38•Updated Feb 22, 2026Feb 22, 2026

    • 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
      •673•4.7k•118•59•Updated Feb 21, 2026Feb 21, 2026

    • 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 Feb 21, 2026Feb 21, 2026

    • FBGEMM

      Public
      FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
      C++
      •
      Other
      •722•1.5k•62•613•Updated Feb 21, 2026Feb 21, 2026

    • vision

      Public
      Datasets, Transforms and Models specific to Computer Vision
      machine-learningcomputer-vision
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •7.2k•18k•913•274•Updated Feb 21, 2026Feb 21, 2026

    • audio

      Public
      Data manipulation and transformation for audio signal processing, powered by PyTorch
      audiopythonmachine-learningspeechpytorchioaudio-processing
      Python
      •
      BSD 2-Clause "Simplified" License
      •766•2.8k•238•81•Updated Feb 21, 2026Feb 21, 2026

    • kineto

      Public
      A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
      HTML
      •
      Other
      •226•922•69•30•Updated Feb 21, 2026Feb 21, 2026

    • pytorch-integration-testing

      Public
      Testing downstream libraries using pytorch release candidates
      Makefile
      •
      Other
      •24•15•1•6•Updated Feb 20, 2026Feb 20, 2026

    • pytorch.github.io

      Public
      The website for PyTorch
      HTML
      •
      BSD 3-Clause "New" or "Revised" License
      •313•276•152•62•Updated Feb 20, 2026Feb 20, 2026

    • tutorials

      Public
      PyTorch tutorials.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •4.4k•9k•204•45•Updated Feb 20, 2026Feb 20, 2026

    • pytorch_sphinx_theme

      Public
      PyTorch Sphinx Theme
      CSS
      •
      MIT License
      •145•105•21•10•Updated Feb 19, 2026Feb 19, 2026

    • 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
      •388•1.2k•48•44•Updated Feb 18, 2026Feb 18, 2026

    • gloo

      Public
      Collective communications library with various primitives for multi-machine training.
      pytorchdistributed-trainingcollectives
      C++
      •
      Other
      •347•1.4k•65•45•Updated Feb 12, 2026Feb 12, 2026

    • alerting-infra

      Public
      Alerting system for the pytorch org
      TypeScript
      •
      BSD 3-Clause "New" or "Revised" License
      •3•2•190•2•Updated Feb 11, 2026Feb 11, 2026

    • ci-infra

      Public
      HCL
      •
      Other
      •7•13•28•1•Updated Feb 9, 2026Feb 9, 2026

    • ort

      Public
      Accelerate PyTorch models with ONNX Runtime
      Python
      •
      MIT License
      •55•368•13•8•Updated Feb 5, 2026Feb 5, 2026

    • extension-cpp

      Public
      C++ extensions in PyTorch
      Python
      •252•1.2k•44•8•Updated Jan 13, 2026Jan 13, 2026

    • xla

      Public
      Enabling PyTorch on XLA Devices (e.g. Google TPU)
      deep-learningcompilerpytorchxla
      C++
      •
      Other
      •565•2.8k•694•185•Updated Dec 18, 2025Dec 18, 2025

    • tensorpipe

      Public
      A tensor-aware point-to-point communication primitive for machine learning
      C++
      •
      Other
      •79•283•47•14•Updated Dec 17, 2025Dec 17, 2025

    • torchchat

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