salesforce / jaxformerLinks
Minimal library to train LLMs on TPU in JAX with pjit().
☆286Updated last year
Alternatives and similar repositories for jaxformer
Users that are interested in jaxformer are comparing it to the libraries listed below
Sorting:
- Fine-tune SantaCoder for Code/Text Generation.☆191Updated 2 years ago
- Code for the paper "Efficient Training of Language Models to Fill in the Middle"☆180Updated 2 years ago
- Generative model for code infilling and synthesis☆302Updated last year
- Repository for analysis and experiments in the BigCode project.☆118Updated last year
- This is the official code for the paper CodeRL: Mastering Code Generation through Pretrained Models and Deep Reinforcement Learning (Neur…☆532Updated 4 months ago
- CodeGen2 models for program synthesis☆272Updated last year
- ☆269Updated 2 years ago
- Scalable PaLM implementation of PyTorch☆189Updated 2 years ago
- GPTQ inference Triton kernel☆300Updated 2 years ago
- Scaling Data-Constrained Language Models☆334Updated 8 months ago
- Fast Inference Solutions for BLOOM☆564Updated 7 months ago
- NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day☆254Updated last year
- Ongoing research training transformer models at scale☆386Updated 9 months ago
- DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.☆166Updated this week
- Used for adaptive human in the loop evaluation of language and embedding models.☆308Updated 2 years ago
- Code for the paper "Rethinking Benchmark and Contamination for Language Models with Rephrased Samples"☆302Updated last year
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆209Updated last year
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 2 years ago
- ☆67Updated 2 years ago
- Multipack distributed sampler for fast padding-free training of LLMs☆188Updated 9 months ago
- This project is an attempt to create a common metric to test LLM's for progress in eliminating hallucinations which is the most serious c…☆222Updated 2 years ago
- Official repository for LongChat and LongEval☆518Updated last year
- Inference code for LLaMA models in JAX☆117Updated last year
- Landmark Attention: Random-Access Infinite Context Length for Transformers☆421Updated last year
- Train very large language models in Jax.☆204Updated last year
- A framework for the evaluation of autoregressive code generation language models.☆946Updated 7 months ago
- Accepted by Transactions on Machine Learning Research (TMLR)☆127Updated 7 months ago
- Build, evaluate, understand, and fix LLM-based apps☆489Updated last year
- ☆455Updated 9 months ago
- DSIR large-scale data selection framework for language model training☆249Updated last year