sanjeevanahilan / nanoChatGPT
A crude RLHF layer on top of nanoGPT with Gumbel-Softmax trick
☆287Updated 11 months ago
Related projects ⓘ
Alternatives and complementary repositories for nanoChatGPT
- Experiments around a simple idea for inducing multiple hierarchical predictive model within a GPT☆205Updated 3 months ago
- JAX implementation of the Llama 2 model☆210Updated 9 months ago
- ☆454Updated last year
- ☆411Updated last year
- Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch☆394Updated this week
- Exploring finetuning public checkpoints on filter 8K sequences on Pile☆115Updated last year
- TART: A plug-and-play Transformer module for task-agnostic reasoning☆190Updated last year
- Language Modeling with the H3 State Space Model☆513Updated last year
- ☆91Updated last year
- batched loras☆336Updated last year
- Ungreedy subword tokenizer and vocabulary trainer for Python, Go & Javascript☆551Updated 4 months ago
- Train very large language models in Jax.☆195Updated last year
- Multipack distributed sampler for fast padding-free training of LLMs☆178Updated 3 months ago
- Fast & Simple repository for pre-training and fine-tuning T5-style models☆970Updated 3 months ago
- Tune any FALCON in 4-bit☆468Updated last year
- Used for adaptive human in the loop evaluation of language and embedding models.☆304Updated last year
- Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate …☆624Updated last year
- A puzzle to learn about prompting☆121Updated last year
- Tools for understanding how transformer predictions are built layer-by-layer☆430Updated 5 months ago
- A repository for research on medium sized language models.☆479Updated this week
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆334Updated 3 months ago
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"☆537Updated 6 months ago
- Code for fine-tuning Platypus fam LLMs using LoRA☆623Updated 9 months ago
- Scaling Data-Constrained Language Models☆321Updated last month
- A minimum example of aligning language models with RLHF similar to ChatGPT☆214Updated last year
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆435Updated 6 months ago
- Inference code for LLaMA models in JAX☆113Updated 6 months ago
- Fine-tune mistral-7B on 3090s, a100s, h100s☆702Updated last year
- Landmark Attention: Random-Access Infinite Context Length for Transformers☆415Updated 11 months ago