convert maxtext trained orbax checkpoint to HF checkpoint

AI-Hypercomputer / maxtext

A simple, performant and scalable Jax LLM!

Apache License 2.0

1.53k stars 293 forks source link

Closed jwyang-google closed 2 months ago

jwyang-google commented 2 months ago

hi @khatwanimohit and @gobbleturk

sorry I messed up with the previous PR and closed that one. Please review this new one instead. The changes are all same.