Pinned Loading

1. qax qax Public

   If it quacks like a tensor...

   Python 52 4

2. lorax lorax Public

   LoRA for arbitrary JAX models and functions

   Python 131 4

3. jax-gptq jax-gptq Public

   JAX implementation of GPTQ quantization algorithm

   Python 8 4

4. haiku-mup haiku-mup Public

   A port of muP to JAX/Haiku

   Python 25 3

5. abnormal-floats abnormal-floats Public

   Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)

   Python 18

6. jax-dropless-moe jax-dropless-moe Public

   WIP implementation of block-sparse dropless MoE in JAX

   Python 2