google / maxtext

A simple, performant and scalable Jax LLM!

Apache License 2.0

1.39k stars 247 forks source link

Closed RissyRan closed 2 weeks ago

RissyRan commented 3 weeks ago

Description

Update custom tile size to a fix one, and get a better MFU and throughput: ~0.6% increase