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•95k•16k•1.6k•Updated Nov 23, 2025Nov 23, 2025

    • TensorRT

      Public
      PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
      machine-learningdeep-learningcudapytorchnvidiajetsontensorrtlibtorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •370•2.9k•275•37•Updated Nov 23, 2025Nov 23, 2025

    • cppdocs

      Public
      PyTorch C++ API Documentation
      HTML
      •37•240•7•3•Updated Nov 23, 2025Nov 23, 2025

    • torchtitan

      Public
      A PyTorch native platform for training generative AI models
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •611•4.7k•143•155•Updated Nov 23, 2025Nov 23, 2025

    • executorch

      Public
      On-device AI across mobile, embedded and edge for PyTorch
      machine-learningmobileembeddeddeep-learningneural-networkgputensor
      Python
      •
      Other
      •732•3.6k•792•676•Updated Nov 23, 2025Nov 23, 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
      •107•103•220•246•Updated Nov 23, 2025Nov 23, 2025

    • ao

      Public
      PyTorch native quantization and sparsity for training and inference
      trainingsparsitycudainferenceoptimizerpytorchtransformeroffloadingllamaquantization
      Python
      •
      Other
      •375•2.5k•348•193•Updated Nov 23, 2025Nov 23, 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
      •6•9•3•0•Updated Nov 23, 2025Nov 23, 2025

    • FBGEMM

      Public
      FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
      C++
      •
      Other
      •686•1.5k•60•585•Updated Nov 23, 2025Nov 23, 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
      •71•635•66•33•Updated Nov 22, 2025Nov 22, 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
      •326•996•81•75•Updated Nov 22, 2025Nov 22, 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•907•251•Updated Nov 22, 2025Nov 22, 2025

    • audio

      Public
      Data manipulation and transformation for audio signal processing, powered by PyTorch
      audiopythonmachine-learningspeechpytorchioaudio-processing
      Python
      •
      BSD 2-Clause "Simplified" License
      •744•2.8k•240•91•Updated Nov 22, 2025Nov 22, 2025

    • xla

      Public
      Enabling PyTorch on XLA Devices (e.g. Google TPU)
      deep-learningcompilerpytorchxla
      C++
      •
      Other
      •561•2.7k•691•185•Updated Nov 21, 2025Nov 21, 2025

    • kineto

      Public
      A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
      HTML
      •
      Other
      •208•894•63•32•Updated Nov 21, 2025Nov 21, 2025

    • pytorch.github.io

      Public
      The website for PyTorch
      HTML
      •
      BSD 3-Clause "New" or "Revised" License
      •314•269•149•64•Updated Nov 20, 2025Nov 20, 2025

    • pytorch-integration-testing

      Public
      Testing downstream libraries using pytorch release candidates
      Makefile
      •
      Other
      •21•13•1•3•Updated Nov 19, 2025Nov 19, 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
      •376•1.1k•47•46•Updated Nov 19, 2025Nov 19, 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
      •656•4.7k•112•40•Updated Nov 18, 2025Nov 18, 2025

    • rl

      Public
      A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
      machine-learningcontrolreinforcement-learningairoboticsdecision-makingdistributed-computingtorchpytorchrl
      Python
      •
      MIT License
      •418•3.2k•208•90•Updated Nov 18, 2025Nov 18, 2025

    • tutorials

      Public
      PyTorch tutorials.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •4.3k•8.9k•185•18•Updated Nov 13, 2025Nov 13, 2025

    • tensorpipe

      Public
      A tensor-aware point-to-point communication primitive for machine learning
      C++
      •
      Other
      •79•274•47•13•Updated Nov 5, 2025Nov 5, 2025

    • pytorch_sphinx_theme

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

    • tensordict

      Public
      TensorDict is a pytorch dedicated tensor container.
      Python
      •
      MIT License
      •105•985•49•38•Updated Oct 25, 2025Oct 25, 2025

    • gloo

      Public
      Collective communications library with various primitives for multi-machine training.
      pytorchdistributed-trainingcollectives
      C++
      •
      Other
      •338•1.4k•63•43•Updated Oct 21, 2025Oct 21, 2025

    • alerting-infra

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

    • ci-infra

      Public
      HCL
      •
      Other
      •4•14•27•0•Updated Sep 23, 2025Sep 23, 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

    • 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
      •814•3.6k•290•56•Updated Sep 10, 2025Sep 10, 2025

    • examples

      Public
      A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •9.8k•24k•198•44•Updated Sep 1, 2025Sep 1, 2025