Pin Flax version in FG+Tunix notebook#328

bebechien merged 2 commits intogoogle-gemma:maingoogle-gemma/cookbook:mainfrom

windmaple:fg-tunixwindmaple/gemma-cookbook:fg-tunixCopy head branch name to clipboard