search

google / maxtext

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

Fix and protect simple_layer #732

Closed gobbleturk closed 3 weeks ago