flax

cc0aada0 - Merge pull request #2901 from google:cgarciae-getattribute-no-recursive-register