Skip to content

@pytorch-labs PyTorch Labs

Change the repository type filter

All

    Repositories list

    24 repositories

    • 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
      •15•195•20•16•Updated Jul 24, 2025Jul 24, 2025

    • monarch

      Public
      PyTorch Single Controller
      Rust
      •
      BSD 3-Clause "New" or "Revised" License
      •51•333•6•84•Updated Jul 24, 2025Jul 24, 2025

    • torchvision-extra-decoders

      Public
      An extension of `torchvision` for decoding AVIF and HEIC images.
      C++
      •
      GNU Lesser General Public License v2.1
      •1•6•2•1•Updated Jul 24, 2025Jul 24, 2025

    • compile-graph-break-site

      Public
      This repository contains the source code for a static website that provides documentation for each "Graph Break" identified by a Graph Break ID (GBID).
      Python
      •
      Other
      •0•0•0•0•Updated Jul 24, 2025Jul 24, 2025

    • executorch-examples

      Public
      Example apps and demos using PyTorch's ExecuTorch framework
      Kotlin
      •
      Other
      •5•15•3•0•Updated Jul 24, 2025Jul 24, 2025

    • tokenizers

      Public
      C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
      C++
      •
      BSD 3-Clause "New" or "Revised" License
      •11•32•5•11•Updated Jul 23, 2025Jul 23, 2025

    • tritonbench

      Public
      Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •26•189•9•9•Updated Jul 23, 2025Jul 23, 2025

    • tritonparse

      Public
      TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer for Triton Kernels
      debuggingcompilerpytorchtritonstructured-logginginteractive-visualizationir-analysisgpu-kernelir-visualization
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •5•134•0•1•Updated Jul 23, 2025Jul 23, 2025

    • torch-release-notes

      Public
      Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.
      BSD 3-Clause "New" or "Revised" License
      •15•1•0•1•Updated Jul 22, 2025Jul 22, 2025

    • chakra_replay

      Public
      Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.
      Python
      •
      Apache License 2.0
      •1•6•0•0•Updated Jul 22, 2025Jul 22, 2025

    • attention-gym

      Public
      Helpful tools and examples for working with flex-attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •55•896•61•5•Updated Jul 18, 2025Jul 18, 2025

    • FACTO

      Public
      Framework for Algorithmic Correctness Testing of Operators
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •4•10•0•0•Updated Jul 14, 2025Jul 14, 2025

    • kraken

      Public
      Triton-based Symmetric Memory operators and examples
      Python
      •
      Other
      •0•6•0•0•Updated Jul 11, 2025Jul 11, 2025

    • test-infra

      Public
      Other
      •4•0•0•0•Updated Jun 24, 2025Jun 24, 2025

    • applied-ai

      Public
      Applied AI experiments and examples for PyTorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •29•287•11•2•Updated May 29, 2025May 29, 2025

    • gpt-fast

      Public
      Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •560•6k•76•42•Updated Apr 11, 2025Apr 11, 2025

    • triton-cpu

      Public
      An experimental CPU backend for Triton (https//github.com/openai/triton)
      C++
      •
      MIT License
      •2•43•1•0•Updated Mar 17, 2025Mar 17, 2025

    • torchfix

      Public
      TorchFix - a linter for PyTorch-using code with autofix support
      pythonstatic-code-analysislinterstatic-analysispytorchflake8hacktoberfestflake8-plugin
      Python
      •
      Other
      •20•145•13•6•Updated Feb 7, 2025Feb 7, 2025

    • tosa_serialization_lib-mirror

      Public
      This is a mirror of https//git.mlplatform.org/tosa/serialization_lib.git/
      C++
      •
      Apache License 2.0
      •0•0•0•0•Updated Jan 15, 2025Jan 15, 2025

    • ethos-u-core-driver-mirror

      Public
      This is a mirror of https//git.mlplatform.org/ml/ethos-u/ethos-u-core-driver.git/
      C++
      •
      Apache License 2.0
      •0•0•0•2•Updated Nov 21, 2024Nov 21, 2024

    • LeanRL

      Public
      LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
      Python
      •
      Other
      •26•606•6•2•Updated Oct 26, 2024Oct 26, 2024

    • segment-anything-fast

      Public
      A batched offline inference oriented version of segment-anything
      Python
      •
      Apache License 2.0
      •71•1.2k•29•4•Updated Sep 13, 2024Sep 13, 2024

    • superblock

      Public archive
      A block oriented training approach for inference time optimization.
      Python
      •
      MIT License
      •4•33•0•1•Updated Aug 19, 2024Aug 19, 2024

    • float8_experimental

      Public archive
      This repository contains the experimental PyTorch native float8 training UX
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •19•224•0•22•Updated Aug 1, 2024Aug 1, 2024