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.5k 503

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

    A batched offline inference oriented version of segment-anything

    Python 1.2k 67

  3. attention-gym attention-gym Public

    Helpful tools and examples for working with flex-attention

    Python 316 13

  4. float8_experimental float8_experimental Public archive

    This repository contains the experimental PyTorch native float8 training UX

    Python 210 20

  5. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 115 8

  6. torchfix torchfix Public

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

    Python 72 16

Repositories

Showing 9 of 9 repositories
  • torchfix Public

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

    pytorch-labs/torchfix’s past year of commit activity
    Python 72 16 10 (3 issues need help) 5 Updated Sep 5, 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,485 BSD-3-Clause 503 64 38 Updated Sep 5, 2024
  • FACTO Public

    Framework for Algorithmic Correctness Testing of Operators

    pytorch-labs/FACTO’s past year of commit activity
    Python 0 BSD-3-Clause 1 0 0 Updated Aug 27, 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,177 Apache-2.0 67 26 6 Updated Aug 23, 2024
  • superblock Public archive

    A block oriented training approach for inference time optimization.

    pytorch-labs/superblock’s past year of commit activity
    Python 25 MIT 3 0 1 Updated Aug 19, 2024
  • applied-ai Public

    Applied AI experiments and examples for PyTorch

    pytorch-labs/applied-ai’s past year of commit activity
    Python 115 BSD-3-Clause 8 8 1 Updated Aug 17, 2024
  • attention-gym Public

    Helpful tools and examples for working with flex-attention

    pytorch-labs/attention-gym’s past year of commit activity
    Python 316 BSD-3-Clause 13 19 (1 issue needs help) 1 Updated Aug 17, 2024
  • float8_experimental Public archive

    This repository contains the experimental PyTorch native float8 training UX

    pytorch-labs/float8_experimental’s past year of commit activity
    Python 210 BSD-3-Clause 20 0 22 Updated Aug 1, 2024
  • triton-cpu Public

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

    pytorch-labs/triton-cpu’s past year of commit activity
    C++ 29 MIT 3 0 0 Updated May 13, 2024

Top languages

Loading…

Most used topics

Loading…