salesforce / jaxformerLinks
Minimal library to train LLMs on TPU in JAX with pjit().
☆290Updated 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.☆119Updated last year
- This is the official code for the paper CodeRL: Mastering Code Generation through Pretrained Models and Deep Reinforcement Learning (Neur…☆534Updated 5 months ago
- Ongoing research training transformer models at scale☆389Updated 10 months ago
- DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.☆167Updated 3 weeks ago
- Simple next-token-prediction for RLHF☆227Updated last year
- The data processing pipeline for the Koala chatbot language model☆117Updated 2 years ago
- CodeGen2 models for program synthesis☆273Updated 2 years ago
- Code for the paper "Efficient Training of Language Models to Fill in the Middle"☆184Updated 2 years ago
- Fine-tune SantaCoder for Code/Text Generation.☆192Updated 2 years ago
- Run evaluation on LLMs using human-eval benchmark☆414Updated last year
- Generative model for code infilling and synthesis☆302Updated last year
- ☆84Updated 2 years ago
- Code for the paper "Rethinking Benchmark and Contamination for Language Models with Rephrased Samples"☆303Updated last year
- Accepted by Transactions on Machine Learning Research (TMLR)☆128Updated 8 months ago
- ☆270Updated 2 years ago
- Scaling Data-Constrained Language Models☆335Updated 9 months ago
- [NeurIPS 2023 D&B] Code repository for InterCode benchmark https://arxiv.org/abs/2306.14898☆220Updated last year
- 🐙 OctoPack: Instruction Tuning Code Large Language Models☆468Updated 4 months ago
- Code for the curation of The Stack v2 and StarCoder2 training data☆108Updated last year
- A framework for the evaluation of autoregressive code generation language models.☆949Updated 7 months ago
- JAX implementation of the Llama 2 model☆218Updated last year
- [ICML 2023] Data and code release for the paper "DS-1000: A Natural and Reliable Benchmark for Data Science Code Generation".☆245Updated 7 months ago
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆277Updated last year
- ☆52Updated 3 months ago
- Train very large language models in Jax.☆204Updated last year
- Data and code for "DocPrompting: Generating Code by Retrieving the Docs" @ICLR 2023☆248Updated last year
- Used for adaptive human in the loop evaluation of language and embedding models.☆309Updated 2 years ago
- ☆110Updated 11 months ago
- This repository contains all the code for collecting large scale amounts of code from GitHub.☆108Updated 2 years ago
- Multipack distributed sampler for fast padding-free training of LLMs☆191Updated 10 months ago