[Qwix/Flax] Upgrade to Flax 0.12.0 + Qwix 0.1.4#1170

jrplatin wants to merge 4 commits intomainvllm-project/tpu-inference:mainfrom

jacobplatin/upgrade-qwixvllm-project/tpu-inference:jacobplatin/upgrade-qwixCopy head branch name to clipboard