flax

f5c48fe8 - Merge pull request #3721 from google:fix-tabulate-flops