flax

f2f6ab54 - Update setup.py

Update setup.py Add tensorflow_text and clu to our test dependencies. tensorflow_text is required for WMT test cases (after #616) and clu will be used in the future.