flax

8070946c - Update setup.py

Update setup.py Also add sentencepiece to tests_require. I created #623 to track the issue that we are using tests_require for all requirements to run the examples.