Fix MGUCell hidden state update#5467

lanarkite99 wants to merge 1 commit into

google:maingoogle/flax:mainfrom

lanarkite99:resume/flax-mgu-cell-fixlanarkite99/flax:resume/flax-mgu-cell-fixCopy head branch name to clipboard