Skip to content
@pytorch-labs

PyTorch Labs

Popular repositories

  1. gpt-fast gpt-fast Public

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

    Python 5.3k 480

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

    A batched offline inference oriented version of segment-anything

    Python 1.1k 66

  3. float8_experimental float8_experimental Public

    This repository contains the experimental PyTorch native float8 training UX

    Python 182 18

  4. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 68 6

  5. torchfix torchfix Public

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

    Python 63 14

  6. triton-cpu triton-cpu Public

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

    C++ 20 3

Repositories

Showing 7 of 7 repositories
  • float8_experimental Public

    This repository contains the experimental PyTorch native float8 training UX

    pytorch-labs/float8_experimental’s past year of commit activity
    Python 182 BSD-3-Clause 18 28 23 Updated Jun 23, 2024
  • superblock Public

    A block oriented training approach for inference time optimization.

    pytorch-labs/superblock’s past year of commit activity
    Python 15 MIT 3 0 1 Updated Jun 21, 2024
  • torchfix Public

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

    pytorch-labs/torchfix’s past year of commit activity
    Python 63 14 9 (3 issues need help) 0 Updated Jun 20, 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,306 BSD-3-Clause 480 61 36 Updated Jun 17, 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,138 Apache-2.0 66 26 5 Updated Jun 5, 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++ 20 MIT 4 0 0 Updated May 13, 2024
  • applied-ai Public

    Applied AI experiments and examples for PyTorch

    pytorch-labs/applied-ai’s past year of commit activity
    Python 68 BSD-3-Clause 6 7 2 Updated May 6, 2024

Top languages

Loading…

Most used topics

Loading…