fork google's maxtext to ROCm and port maxtext, please make the forked repository private - View it on GitHub