Popular repositories Loading

1. gpt-fast gpt-fast Public

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

   Python 5.6k 509

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

   A batched offline inference oriented version of segment-anything

   Python 1.2k 70

3. LeanRL LeanRL Public

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

   Python 407 16

4. attention-gym attention-gym Public

   Helpful tools and examples for working with flex-attention

   Python 386 19

5. float8_experimental float8_experimental Public archive

   This repository contains the experimental PyTorch native float8 training UX

   Python 212 20

6. applied-ai applied-ai Public

   Applied AI experiments and examples for PyTorch

   Python 144 12