flax

35311878 - Merge pull request #5056 from chapman20j:just_set_mode