flax

a31129e2 - Merge pull request #5526 from samanklesaria:fix_ci_precision