* Update requirements for TPU support.

This includes a complete installation as per the instructions in
https://github.com/huggingface/transformers/tree/master/examples/research_projects/jax-projects#tpu-vm

Note that `transformers` and `datasets` have been installed from source,
in editable mode.

requirements.txt

@@ -1,5 +1,102 @@
-jax
-flax
-optax
-datasets
-git+https://github.com/huggingface/transformers
+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