search

AI-Hypercomputer / maxtext

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

[WIP] partial nnx impl #896

Open rdyro opened 1 month ago