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.7k 514

  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 462 23

  4. LeanRL LeanRL Public

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

    Python 432 17

  5. float8_experimental float8_experimental Public archive

    This repository contains the experimental PyTorch native float8 training UX

    Python 211 20

  6. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 160 13

Repositories

Showing 10 of 13 repositories
  • torchft Public

    PyTorch per step fault tolerance (actively under development)

    pytorch-labs/torchft’s past year of commit activity
    Python 4 0 0 1 Updated Nov 9, 2024
  • 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 15 BSD-3-Clause 3 6 2 Updated Nov 9, 2024
  • applied-ai Public

    Applied AI experiments and examples for PyTorch

    pytorch-labs/applied-ai’s past year of commit activity
    Python 160 BSD-3-Clause 13 8 0 Updated Oct 31, 2024
  • test-infra Public
    pytorch-labs/test-infra’s past year of commit activity
    0 1 0 1 Updated Oct 29, 2024
  • LeanRL Public

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

    pytorch-labs/LeanRL’s past year of commit activity
    Python 432 17 4 1 Updated Oct 26, 2024
  • attention-gym Public

    Helpful tools and examples for working with flex-attention

    pytorch-labs/attention-gym’s past year of commit activity
    Python 462 BSD-3-Clause 23 37 (1 issue needs help) 1 Updated Oct 23, 2024
  • gpt-fast Public

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

    pytorch-labs/gpt-fast’s past year of commit activity
    Python 5,659 BSD-3-Clause 514 66 42 Updated Oct 18, 2024
  • torchfix Public

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

    pytorch-labs/torchfix’s past year of commit activity
    Python 100 16 10 (3 issues need help) 4 Updated Sep 24, 2024
  • segment-anything-fast Public

    A batched offline inference oriented version of segment-anything

    pytorch-labs/segment-anything-fast’s past year of commit activity
    Python 1,197 Apache-2.0 71 27 5 Updated Sep 14, 2024
  • FACTO Public

    Framework for Algorithmic Correctness Testing of Operators

    pytorch-labs/FACTO’s past year of commit activity
    Python 1 BSD-3-Clause 1 0 0 Updated Aug 27, 2024
View all repositories

Top languages

Python C++

Most used topics

Loading…