flax

f9f68854 - Merge pull request #4554 from google:bridge-improve-stack