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.☆120Updated last year
- Code for the paper "Efficient Training of Language Models to Fill in the Middle"☆183Updated 2 years ago
- This is the official code for the paper CodeRL: Mastering Code Generation through Pretrained Models and Deep Reinforcement Learning (Neur…☆536Updated 5 months ago
- CodeGen2 models for program synthesis☆273Updated 2 years ago
- The data processing pipeline for the Koala chatbot language model☆117Updated 2 years ago
- Fine-tune SantaCoder for Code/Text Generation.☆192Updated 2 years ago
- DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.☆168Updated last month
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆209Updated last year
- Generative model for code infilling and synthesis☆303Updated last year
- Ongoing research training transformer models at scale☆389Updated 10 months ago
- Inference code for LLaMA models in JAX☆118Updated last year
- Code for the paper "Rethinking Benchmark and Contamination for Language Models with Rephrased Samples"☆306Updated last year
- Scaling Data-Constrained Language Models☆337Updated 2 weeks ago
- ☆270Updated 2 years ago
- Accepted by Transactions on Machine Learning Research (TMLR)☆129Updated 9 months ago
- Multipack distributed sampler for fast padding-free training of LLMs☆194Updated 11 months ago
- Simple next-token-prediction for RLHF☆228Updated last year
- This is the repo for the paper Shepherd -- A Critic for Language Model Generation☆219Updated last year
- batched loras☆343Updated last year
- ☆256Updated last month
- A hard gym for programming☆159Updated last year
- Fast Inference Solutions for BLOOM☆564Updated 9 months ago
- NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day☆256Updated last year
- ☆67Updated 2 years ago
- Train very large language models in Jax.☆204Updated last year
- Salesforce open-source LLMs with 8k sequence length.☆718Updated 5 months ago
- APPS: Automated Programming Progress Standard (NeurIPS 2021)☆472Updated last year
- Landmark Attention: Random-Access Infinite Context Length for Transformers☆423Updated last year
- Task-based datasets, preprocessing, and evaluation for sequence models.☆583Updated 2 months ago
- ☆464Updated 11 months ago