salesforce / jaxformerLinks
Minimal library to train LLMs on TPU in JAX with pjit().
☆298Updated last year
Alternatives and similar repositories for jaxformer
Users that are interested in jaxformer are comparing it to the libraries listed below
Sorting:
- Repository for analysis and experiments in the BigCode project.☆125Updated last year
- Code for the paper "Efficient Training of Language Models to Fill in the Middle"☆194Updated 2 years ago
- Fine-tune SantaCoder for Code/Text Generation.☆193Updated 2 years ago
- CodeGen2 models for program synthesis☆269Updated 2 years ago
- DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.☆170Updated last month
- This is the official code for the paper CodeRL: Mastering Code Generation through Pretrained Models and Deep Reinforcement Learning (Neur…☆552Updated 9 months ago
- Generative model for code infilling and synthesis☆308Updated 2 years ago
- Ongoing research training transformer models at scale☆391Updated last year
- Inference code for LLaMA models in JAX☆119Updated last year
- Used for adaptive human in the loop evaluation of language and embedding models.☆307Updated 2 years ago
- This repository contains all the code for collecting large scale amounts of code from GitHub.☆109Updated 2 years ago
- Code for the paper "Rethinking Benchmark and Contamination for Language Models with Rephrased Samples"☆312Updated last year
- Train very large language models in Jax.☆210Updated 2 years ago
- The data processing pipeline for the Koala chatbot language model☆118Updated 2 years ago
- Scaling Data-Constrained Language Models☆341Updated 4 months ago
- Accepted by Transactions on Machine Learning Research (TMLR)☆133Updated last year
- A crude RLHF layer on top of nanoGPT with Gumbel-Softmax trick☆293Updated last year
- Multipack distributed sampler for fast padding-free training of LLMs☆201Updated last year
- ☆274Updated 2 years ago
- This is the repo for the paper Shepherd -- A Critic for Language Model Generation☆217Updated 2 years ago
- ☆476Updated last year
- batched loras☆347Updated 2 years ago
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆209Updated last year
- A hard gym for programming☆160Updated last year
- Code for the curation of The Stack v2 and StarCoder2 training data☆120Updated last year
- ☆83Updated 2 years ago
- Merge Transformers language models by use of gradient parameters.☆208Updated last year
- A repository for research on medium sized language models.☆518Updated 5 months ago
- Run evaluation on LLMs using human-eval benchmark☆422Updated 2 years ago
- Simple next-token-prediction for RLHF☆226Updated 2 years ago