salesforce / jaxformerLinks
Minimal library to train LLMs on TPU in JAX with pjit().
☆296Updated last year
Alternatives and similar repositories for jaxformer
Users that are interested in jaxformer are comparing it to the libraries listed below
Sorting:
- Code for the paper "Efficient Training of Language Models to Fill in the Middle"☆183Updated 2 years ago
- Repository for analysis and experiments in the BigCode project.☆123Updated last year
- 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
- ☆272Updated 2 years ago
- Ongoing research training transformer models at scale☆392Updated last year
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆208Updated last year
- This is the official code for the paper CodeRL: Mastering Code Generation through Pretrained Models and Deep Reinforcement Learning (Neur…☆541Updated 7 months ago
- Generative model for code infilling and synthesis☆306Updated last year
- The data processing pipeline for the Koala chatbot language model☆118Updated 2 years ago
- Multipack distributed sampler for fast padding-free training of LLMs☆199Updated last year
- Train very large language models in Jax.☆206Updated last year
- ☆52Updated 5 months ago
- Scaling Data-Constrained Language Models☆339Updated 2 months ago
- CodeGen2 models for program synthesis☆271Updated 2 years ago
- Inference code for LLaMA models in JAX☆118Updated last year
- ☆67Updated 3 years ago
- Accepted by Transactions on Machine Learning Research (TMLR)☆130Updated 10 months ago
- JAX implementation of the Llama 2 model☆219Updated last year
- Minimal code to train a Large Language Model (LLM).☆172Updated 3 years ago
- Code for the paper "Rethinking Benchmark and Contamination for Language Models with Rephrased Samples"☆309Updated last year
- batched loras☆345Updated last year
- Landmark Attention: Random-Access Infinite Context Length for Transformers☆423Updated last year
- Simple next-token-prediction for RLHF☆227Updated last year
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 3 years ago
- This repository contains all the code for collecting large scale amounts of code from GitHub.☆110Updated 2 years ago
- Data and code for "DocPrompting: Generating Code by Retrieving the Docs" @ICLR 2023☆249Updated last year
- Code for the curation of The Stack v2 and StarCoder2 training data☆113Updated last year
- Run evaluation on LLMs using human-eval benchmark☆419Updated last year
- 🐙 OctoPack: Instruction Tuning Code Large Language Models☆471Updated 6 months ago