google / maxtext

A simple, performant and scalable Jax LLM!
Apache License 2.0
1.44k stars 263 forks source link

Update MaxText config for Llama2 7B on GPUs. #704

Open yangyuwei opened 2 months ago