Pinned

1. JAXSeq JAXSeq Public

   Train very large language models in Jax.

   Python 183 17

2. Implicit-Language-Q-Learning Implicit-Language-Q-Learning Public

   Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"

   Python 191 16

3. JAX_llama JAX_llama Public

   Forked from meta-llama/llama

   Inference code for LLaMA models in JAX

   Python 101 6

4. grokking grokking Public

   unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"

   Python 47 14

5. MLLibCpp MLLibCpp Public

   A machine learning library capable of training various deep neural networks (RNNs, LSTMs, DBNs, ect...) on a GPU. It makes use of auto-differentiation algorithms. Written in C++ with OpenCl.

   C++ 9 2

6. 2d-procedural-endless-world 2d-procedural-endless-world Public

   2d procedural endless world

   Swift 4