Pinned Loading

1. EleutherAI/lm-evaluation-harness EleutherAI/lm-evaluation-harness Public

   A framework for few-shot evaluation of language models.

   Python 7.2k 1.9k

2. transformer_shmap transformer_shmap Public

   Tensor Parallelism with JAX + Shard Map

   Python 11 1

3. ZeRO-transformer ZeRO-transformer Public

   Two implementations of ZeRO-1 optimizer sharding in JAX

   Python 13

4. Flax-ResNets Flax-ResNets Public

   CIFAR10 ResNets implemented in JAX+Flax

   Python 9 1

5. fast_sequential_scan fast_sequential_scan Public

   A fast sequential scan on GPU

   Cuda

6. tritonformer tritonformer Public

   Differentiable transformer in Triton, matching the performance of PyTorch + cuDNN/cuBLAS

   Python 2