Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
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 519

  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 500 25

  4. LeanRL LeanRL Public

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

    Python 466 17

  5. float8_experimental float8_experimental Public archive

    This repository contains the experimental PyTorch native float8 training UX

    Python 213 20

  6. applied-ai applied-ai Public

    Applied AI experiments and examples for PyTorch

    Python 176 15

Repositories

Showing 10 of 17 repositories
  • 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 25 BSD-3-Clause 3 14 5 Updated Dec 7, 2024
  • torchft Public

    PyTorch per step fault tolerance (actively under development)

    pytorch-labs/torchft’s past year of commit activity
    Python 9 1 0 1 Updated Dec 7, 2024
  • tokenizers Public

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

    pytorch-labs/tokenizers’s past year of commit activity
    C++ 2 BSD-3-Clause 0 0 0 Updated Dec 6, 2024
  • torchvision-extra-decoders Public

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

    pytorch-labs/torchvision-extra-decoders’s past year of commit activity
    C++ 3 LGPL-2.1 1 0 2 Updated Dec 5, 2024
  • applied-ai Public

    Applied AI experiments and examples for PyTorch

    pytorch-labs/applied-ai’s past year of commit activity
    Python 176 BSD-3-Clause 15 8 1 Updated Dec 4, 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 Dec 4, 2024
  • attention-gym Public

    Helpful tools and examples for working with flex-attention

    pytorch-labs/attention-gym’s past year of commit activity
    Python 500 BSD-3-Clause 25 39 1 Updated Dec 3, 2024
  • ethos-u-core-driver-mirror Public

    This is a mirror of https//git.mlplatform.org/ml/ethos-u/ethos-u-core-driver.git/

    pytorch-labs/ethos-u-core-driver-mirror’s past year of commit activity
    C++ 0 Apache-2.0 0 0 2 Updated Nov 21, 2024
  • tosa_serialization_lib-mirror Public

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

    pytorch-labs/tosa_serialization_lib-mirror’s past year of commit activity
    C++ 0 Apache-2.0 0 0 2 Updated Nov 21, 2024
  • torchfix Public

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

    pytorch-labs/torchfix’s past year of commit activity
    Python 108 16 10 (2 issues need help) 4 Updated Nov 15, 2024
View all repositories

Top languages

Loading…

Most used topics

Loading…