installing one by one, first chex

requirements.txt

@@ -1,133 +1 @@
-absl-py==2.0.0
-accelerate==0.24.1
-aiohttp==3.9.1
-aiosignal==1.3.1
-altair==5.2.0
-antlr4-python3-runtime==4.9.1
-appdirs==1.4.4
-asttokens==2.4.1
-async-timeout==4.0.3
-attrs==23.1.0
-beartype==0.14.1
-blinker==1.7.0
-cachetools==5.3.2
-certifi==2023.11.17
-charset-normalizer==3.3.2
-chex==0.1.85
-click==8.1.7
-colorama==0.4.6
-comm==0.2.0
-contourpy==1.2.0
-cycler==0.12.1
-datasets==2.15.0
-debugpy==1.8.0
-decorator==5.1.1
-dill==0.3.7
-dm-haiku==0.0.11
-docker-pycreds==0.4.0
-einops==0.7.0
-etils==1.5.2
-exceptiongroup==1.2.0
-executing==2.0.1
-fancy-einsum==0.0.3
-fastjsonschema==2.19.0
-filelock==3.13.1
-flax==0.7.5
-fonttools==4.46.0
-frozenlist==1.4.0
-fsspec==2023.10.0
-gitdb==4.0.11
-GitPython==3.1.40
-graphviz==0.20.1
-huggingface-hub==0.19.4
-idna==3.6
-importlib-metadata==6.8.0
-importlib-resources==6.1.1
-ipykernel==6.26.0
-ipython==8.18.0
-jax==0.4.20
-jaxlib==0.4.20
-jaxtyping==0.2.23
-jedi==0.19.1
-Jinja2==3.1.2
-jmp==0.0.4
-jsonschema==4.20.0
-jsonschema-specifications==2023.11.1
-jupyter_client==8.6.0
-jupyter_core==5.5.0
-kiwisolver==1.4.5
-markdown-it-py==3.0.0
-MarkupSafe==2.1.3
-matplotlib==3.8.2
-matplotlib-inline==0.1.6
-mdurl==0.1.2
-ml-dtypes==0.3.1
-mpmath==1.3.0
-msgpack==1.0.7
-multidict==6.0.4
-multiprocess==0.70.15
-nbformat==5.9.2
-nest-asyncio==1.5.8
-networkx==3.2.1
-numpy==1.26.2
-opt-einsum==3.3.0
-optax==0.1.7
-orbax-checkpoint==0.4.3
-packaging==23.2
-pandas==2.1.3
-parso==0.8.3
-Pillow==10.1.0
-platformdirs==4.0.0
-plotly==5.18.0
-prompt-toolkit==3.0.41
-protobuf==4.25.1
-psutil==5.9.6
-pure-eval==0.2.2
-pyarrow==14.0.1
-pyarrow-hotfix==0.6
-pydeck==0.8.1b0
-Pygments==2.17.2
-pyparsing==3.1.1
-python-dateutil==2.8.2
-pytz==2023.3.post1
-PyYAML==6.0.1
-pyzmq==25.1.1
-referencing==0.31.0
-regex==2023.10.3
-requests==2.31.0
-rich==13.7.0
-rpds-py==0.13.1
-safetensors==0.4.0
-scipy==1.11.4
-sentry-sdk==1.37.1
-setproctitle==1.3.3
-six==1.16.0
-smmap==5.0.1
-stack-data==0.6.3
-streamlit==1.29.0
-sympy==1.12
-tabulate==0.9.0
-tenacity==8.2.3
-tensorstore==0.1.50
-tokenizers==0.15.0
-toml==0.10.2
-toolz==0.12.0
-torch==2.1.1
-tornado==6.3.3
-tqdm==4.66.1
-tracr==1.0.0
-traitlets==5.13.0
-transformer-lens==1.10.0
-transformers==4.35.2
-typeguard==2.13.3
-typing_extensions==4.8.0
-tzdata==2023.3
-tzlocal==5.2
-urllib3==2.1.0
-validators==0.22.0
-wandb==0.16.0
-watchdog==3.0.0
-wcwidth==0.2.12
-xxhash==3.4.1
-yarl==1.9.3
-zipp==3.17.0
+chex