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.
google/flax,google/flax