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 545

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

    A batched offline inference oriented version of segment-anything

    Python 1.2k 72

  3. attention-gym attention-gym Public

    Helpful tools and examples for working with flex-attention

    Python 695 37

  4. LeanRL LeanRL Public

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

    Python 525 21

  5. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 250 24

  6. float8_experimental float8_experimental Public archive

    This repository contains the experimental PyTorch native float8 training UX

    Python 222 19

Repositories

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

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

    C++ 3 LGPL-2.1 1 1 1 Updated Mar 23, 2025
  • tritonbench Public

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

    Python 104 BSD-3-Clause 12 16 7 Updated Mar 21, 2025
  • applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 250 BSD-3-Clause 24 10 3 Updated Mar 21, 2025
  • tokenizers Public

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

    C++ 19 BSD-3-Clause 3 0 0 Updated Mar 18, 2025
  • attention-gym Public

    Helpful tools and examples for working with flex-attention

    Python 695 BSD-3-Clause 37 58 3 Updated Mar 18, 2025
  • triton-cpu Public

    An experimental CPU backend for Triton (https//github.com/openai/triton)

    C++ 39 MIT 2 1 0 Updated Mar 17, 2025
  • gpt-fast Public

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

    Python 5,893 BSD-3-Clause 545 71 42 Updated Mar 13, 2025
  • torchfix Public

    TorchFix - a linter for PyTorch-using code with autofix support

    Python 136 20 11 (2 issues need help) 6 Updated Feb 7, 2025
  • FACTO Public

    Framework for Algorithmic Correctness Testing of Operators

    Python 3 BSD-3-Clause 1 0 0 Updated Jan 24, 2025
  • tosa_serialization_lib-mirror Public

    This is a mirror of https//git.mlplatform.org/tosa/serialization_lib.git/

    C++ 0 Apache-2.0 0 0 0 Updated Jan 15, 2025
View all repositories

Top languages

Python C++