QLoRA Support + NNX Decoder Sharding Fixes#3968

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

jackyf/qlora-supportAI-Hypercomputer/maxtext:jackyf/qlora-supportCopy head branch name to clipboard