young-geng/EasyLM

This is stars and forks stats for /young-geng/EasyLM repository. As of 25 Apr, 2024 this repository has 1976 stars and 191 forks.

EasyLM Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax. EasyLM can scale up LLM training to hundreds of TPU/GPU accelerators by leveraging JAX's pjit functionality. Building on top of Hugginface's transformers and datasets, this repo provides an easy to use and easy to customize codebase for training large language models without the complexity in many other frameworks. EasyLM is built with JAX/Flax. By leveraging...