flax

aa9d5728 - Merge pull request #4627 from google:remove-deprecated