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 5.9k 551

  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 777 45

  4. LeanRL LeanRL Public

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

    Python 570 26

  5. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 267 27

  6. float8_experimental float8_experimental Public archive

    This repository contains the experimental PyTorch native float8 training UX

    Python 223 19

Repositories

Showing 10 of 20 repositories
  • torchvision-extra-decoders Public

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

    C++ 5 LGPL-2.1 1 1 1 Updated May 15, 2025
  • helion Public

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

    Python 135 BSD-3-Clause 7 0 2 Updated May 15, 2025
  • tritonparse Public

    TritonParse is a new tool designed to help developers analyze and debug Triton kernels by visualizing the compilation process and source code mappings.

    Python 14 BSD-3-Clause 0 0 0 Updated May 15, 2025
  • tokenizers Public

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

    C++ 22 BSD-3-Clause 7 3 6 Updated May 14, 2025
  • tritonbench Public

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

    Python 124 BSD-3-Clause 17 11 4 Updated May 13, 2025
  • attention-gym Public

    Helpful tools and examples for working with flex-attention

    Python 777 BSD-3-Clause 45 61 3 Updated May 5, 2025
  • applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 267 BSD-3-Clause 27 11 2 Updated Apr 28, 2025
  • executorch-examples Public

    Example apps and demos using PyTorch's ExecuTorch framework

    Swift 6 0 1 0 Updated Apr 25, 2025
  • test-infra Public
    0 3 0 0 Updated Apr 24, 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.

    0 BSD-3-Clause 0 0 0 Updated Apr 23, 2025
View all repositories

Top languages

Loading…

Most used topics

Loading…