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.2k 463

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

    A batched offline inference oriented version of segment-anything

    Python 1.1k 63

  3. float8_experimental float8_experimental Public

    This repository contains the experimental PyTorch native float8 training UX

    Python 170 18

  4. torchfix torchfix Public

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

    Python 60 7

  5. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 59 5

  6. triton-cpu triton-cpu Public

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

    C++ 16 4

Repositories

Showing 7 of 7 repositories

Top languages

Loading…

Most used topics

Loading…