* Simplify requirements.txt, do not pin packages.

Note, however, that you need to install using the following command:

```
$ pip install -r requirements.txt -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
```

Otherwise, pip won't be able to find the appropriate libtpu_nightly.

requirements.txt

@@ -1,102 +1,9 @@
-absl-py==0.13.0
-aiohttp==3.7.4.post0
-argon2-cffi==20.1.0
-async-generator==1.10
-async-timeout==3.0.1
-attrs==21.2.0
-backcall==0.2.0
-bleach==3.3.0
-certifi==2021.5.30
-cffi==1.14.5
-chardet==4.0.0
-chex==0.0.8
-click==8.0.1
-cycler==0.10.0
--e git+https://github.com/huggingface/datasets.git@8c80e4390532cae2ec46c31e5081db6fa3bce541#egg=datasets
-debugpy==1.3.0
-decorator==5.0.9
-defusedxml==0.7.1
-dill==0.3.4
-dm-tree==0.1.6
-entrypoints==0.3
-filelock==3.0.12
-flatbuffers==2.0
-flax==0.3.4
-fsspec==2021.6.1
-huggingface-hub==0.0.12
-idna==2.10
-ipykernel==6.0.1
-ipython==7.25.0
-ipython-genutils==0.2.0
-ipywidgets==7.6.3
-jax==0.2.16
-jaxlib==0.1.68
-jedi==0.18.0
-Jinja2==3.0.1
-joblib==1.0.1
-jsonschema==3.2.0
-jupyter==1.0.0
-jupyter-client==6.1.12
-jupyter-console==6.4.0
-jupyter-core==4.7.1
-jupyterlab-pygments==0.1.2
-jupyterlab-widgets==1.0.0
-kiwisolver==1.3.1
-libtpu-nightly==0.1.dev20210615
-MarkupSafe==2.0.1
-matplotlib==3.4.2
-matplotlib-inline==0.1.2
-mistune==0.8.4
-msgpack==1.0.2
-multidict==5.1.0
-multiprocess==0.70.12.2
-nbclient==0.5.3
-nbconvert==6.1.0
-nbformat==5.1.3
-nest-asyncio==1.5.1
-notebook==6.4.0
-numpy==1.21.0
-opt-einsum==3.3.0
-optax==0.0.9
-packaging==21.0
-pandas==1.3.0
-pandocfilters==1.4.3
-parso==0.8.2
-pexpect==4.8.0
-pickleshare==0.7.5
-Pillow==8.3.0
-prometheus-client==0.11.0
-prompt-toolkit==3.0.19
-ptyprocess==0.7.0
-pyarrow==4.0.1
-pycparser==2.20
-Pygments==2.9.0
-pyparsing==2.4.7
-pyrsistent==0.18.0
-python-dateutil==2.8.1
-pytz==2021.1
-PyYAML==5.4.1
-pyzmq==22.1.0
-qtconsole==5.1.1
-QtPy==1.9.0
-regex==2021.7.1
-requests==2.25.1
-sacremoses==0.0.45
-scipy==1.7.0
-Send2Trash==1.7.1
-six==1.16.0
-terminado==0.10.1
-testpath==0.5.0
-tokenizers==0.10.3
-toolz==0.11.1
-tornado==6.1
-tqdm==4.61.1
-traitlets==5.0.5
--e git+https://github.com/pcuenca/transformers.git@f1c81d6b921fae7adf0f80be7e2567e92221ab1f#egg=transformers
-typing-extensions==3.10.0.0
-urllib3==1.26.6
-wcwidth==0.2.5
-webencodings==0.5.1
-widgetsnbextension==3.5.1
-xxhash==2.0.2
-yarl==1.6.3
+# Note: install with the following command:
+# pip install -r requirements.txt -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+# Otherwise it won't find the appropriate libtpu_nightly
+requests
+jax[tpu]>=0.2.16
+-e git+https://github.com/huggingface/transformers.git@master#egg=transformers
+-e git+https://github.com/huggingface/datasets.git@master#egg=datasets
+flax
+jupyter