math/py-flax: Update to 0.6.8

Changes: https://github.com/google/flax/releases