Skip to content
@pytorch-labs

PyTorch Labs

Popular repositories Loading

  1. gpt-fast gpt-fast Public

    Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

    Python 6k 561

  2. segment-anything-fast segment-anything-fast Public

    A batched offline inference oriented version of segment-anything

    Python 1.2k 71

  3. attention-gym attention-gym Public

    Helpful tools and examples for working with flex-attention

    Python 904 57

  4. LeanRL LeanRL Public

    LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.

    Python 609 26

  5. monarch monarch Public

    PyTorch Single Controller

    Rust 344 52

  6. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 289 29

Repositories

Showing 10 of 24 repositories
  • tritonbench Public

    Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.

    pytorch-labs/tritonbench’s past year of commit activity
    Python 197 BSD-3-Clause 30 8 9 Updated Aug 4, 2025
  • monarch Public

    PyTorch Single Controller

    pytorch-labs/monarch’s past year of commit activity
    Rust 344 BSD-3-Clause 52 8 102 Updated Aug 3, 2025
  • torchvision-extra-decoders Public

    An extension of `torchvision` for decoding AVIF and HEIC images.

    pytorch-labs/torchvision-extra-decoders’s past year of commit activity
    C++ 6 LGPL-2.1 1 2 1 Updated Aug 3, 2025
  • helion Public

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

    pytorch-labs/helion’s past year of commit activity
    Python 203 BSD-3-Clause 16 21 14 Updated Aug 2, 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).

    pytorch-labs/compile-graph-break-site’s past year of commit activity
    Python 0 0 0 0 Updated Aug 1, 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.

    pytorch-labs/torch-release-notes’s past year of commit activity
    Python 1 BSD-3-Clause 15 0 0 Updated Aug 1, 2025
  • FACTO Public

    Framework for Algorithmic Correctness Testing of Operators

    pytorch-labs/FACTO’s past year of commit activity
    Python 10 BSD-3-Clause 5 0 0 Updated Aug 1, 2025
  • tritonparse Public

    TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer(WIP) for Triton Kernels

    pytorch-labs/tritonparse’s past year of commit activity
    Python 138 BSD-3-Clause 7 0 1 Updated Aug 1, 2025
  • tokenizers Public

    C++ implementations for various tokenizers (sentencepiece, tiktoken etc).

    pytorch-labs/tokenizers’s past year of commit activity
    C++ 34 BSD-3-Clause 11 5 13 Updated Aug 1, 2025
  • chakra_replay Public

    Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.

    pytorch-labs/chakra_replay’s past year of commit activity
    Python 6 Apache-2.0 2 0 1 Updated Jul 31, 2025
View all repositories

Most used topics

Loading…