Popular repositories Loading

1. gpt-fast gpt-fast Public

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

   Python 6.1k 565

2. opacus opacus Public

   Training PyTorch models with differential privacy

   Jupyter Notebook 1.9k 378

3. segment-anything-fast segment-anything-fast Public

   A batched offline inference oriented version of segment-anything

   Python 1.2k 73

4. attention-gym attention-gym Public

   Helpful tools and examples for working with flex-attention

   Python 980 60

5. LeanRL LeanRL Public

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

   Python 631 28

6. monarch monarch Public

   PyTorch Single Controller

   Rust 420 73