----------------- Module Specific Help for "jax/0.4.23-py3.9" ------------------ JAX is Autograd and XLA, brought together for high-performance machine learning research. JAX 0.4.23 library for Python 3.9 and CUDA 12.2 For more help see: https://github.com/google/jax NOTE: This software runs in an Apptainer (Singularity) container. Wrapper scripts have been provided so that common commands such as python, python3, pip and pip3 should work as normal. For more information, see: https://docs.csc.fi/computing/containers/run-existing/ Package Version ---------------------------- --------------------- absl-py 2.1.0 aiohttp 3.9.3 aiohttp-cors 0.7.0 aiorwlock 1.4.0 aiosignal 1.3.1 ale-py 0.8.1 annotated-types 0.6.0 anyio 4.2.0 argon2-cffi 23.1.0 argon2-cffi-bindings 21.2.0 arrow 1.3.0 asttokens 2.4.1 astunparse 1.6.3 async-lru 2.0.4 async-timeout 4.0.3 attrs 23.2.0 AutoROM 0.4.2 AutoROM.accept-rom-license 0.6.1 Babel 2.14.0 bayesian-optimization 1.4.3 beautifulsoup4 4.12.3 bleach 6.1.0 blessed 1.20.0 blosc2 2.5.1 box2d-py 2.3.5 cachetools 5.3.2 certifi 2024.2.2 cffi 1.16.0 cftime 1.6.3 charset-normalizer 3.3.2 chex 0.1.85 click 8.1.7 cloudpickle 3.0.0 coax 0.1.13 colorama 0.4.6 colorful 0.5.6 comm 0.2.1 ConfigSpace 0.7.1 contextlib2 21.6.0 contourpy 1.2.0 cycler 0.12.1 Cython 3.0.8 dask 2024.2.0 dask-jobqueue 0.8.2 dataclasses 0.6 datasets 2.17.0 debugpy 1.8.1 decorator 5.1.1 defusedxml 0.7.1 dill 0.3.8 distlib 0.3.8 distrax 0.1.5 distributed 2024.2.0 dm-control 1.0.16 dm-env 1.6 dm-haiku 0.0.11 dm-tree 0.1.8 e3nn-jax 0.20.6 einops 0.7.0 entrypoints 0.4 equinox 0.11.3 et-xmlfile 1.1.0 etils 1.5.2 evaluate 0.4.1 exceptiongroup 1.2.0 executing 2.0.1 Farama-Notifications 0.0.4 fastapi 0.108.0 fastjsonschema 2.19.1 filelock 3.13.1 flatbuffers 23.5.26 flax 0.8.1 fonttools 4.48.1 fqdn 1.5.1 frozenlist 1.4.1 fsspec 2023.10.0 future 0.18.3 gast 0.5.4 glfw 2.6.5 google-api-core 2.17.0 google-auth 2.27.0 google-auth-oauthlib 1.2.0 google-pasta 0.2.0 googleapis-common-protos 1.62.0 gpustat 1.1.1 graphviz 0.20.1 grpcio 1.60.1 gymnasium 0.29.1 h11 0.14.0 h5py 3.10.0 hpbandster 0.7.4 httpcore 1.0.2 httptools 0.6.1 httpx 0.26.0 huggingface-hub 0.20.3 hyperopt 0.2.7 idna 3.6 imageio 2.34.0 imbalanced-learn 0.12.0 importlib-metadata 4.13.0 importlib-resources 6.1.1 iniconfig 2.0.0 ipykernel 6.29.2 ipython 8.18.1 ipython-genutils 0.2.0 isoduration 20.11.0 jax 0.4.23 jax-md 0.2.8 jaxlib 0.4.23+cuda12.cudnn89 jaxtyping 0.2.25 jedi 0.19.1 Jinja2 3.1.3 jmp 0.0.4 joblib 1.3.2 jraph 0.0.6.dev0 json5 0.9.14 jsonpointer 2.4 jsonschema 4.21.1 jsonschema-specifications 2023.12.1 jupyter-client 8.6.0 jupyter-core 5.7.1 jupyter-events 0.9.0 jupyter-lsp 2.2.2 jupyter-server 2.12.5 jupyter-server-terminals 0.5.2 jupyterlab 4.1.2 jupyterlab-pygments 0.3.0 jupyterlab-server 2.25.2 keras 2.15.0 kiwisolver 1.4.5 labmaze 1.0.6 lazy-loader 0.3 libclang 16.0.6 llvmlite 0.42.0 lmdb 1.4.1 locket 1.0.0 lxml 5.1.0 lz4 4.3.3 Markdown 3.5.2 markdown-it-py 3.0.0 MarkupSafe 2.1.5 matplotlib 3.8.2 matplotlib-inline 0.1.6 mdurl 0.1.2 mistune 3.0.2 ml-collections 0.1.1 ml-dtypes 0.2.0 more-itertools 10.2.0 mpmath 1.3.0 msgpack 1.0.7 mujoco 3.1.2 multidict 6.0.5 multiprocess 0.70.16 mysql-connector-python 8.3.0 nbclassic 1.0.0 nbclient 0.9.0 nbconvert 7.16.0 nbformat 5.9.2 ndindex 1.7 nest-asyncio 1.6.0 netCDF4 1.6.5 netifaces 0.11.0 networkx 3.2.1 notebook-shim 0.2.3 numba 0.59.0 numexpr 2.9.0 numpy 1.26.4 nvidia-dlprof 1.8.0 nvidia-ml-py 12.535.133 nvidia-nsys-cli 2021.3.2.12 nvidia-pyindex 1.0.9 oauthlib 3.2.2 objax 1.8.0 odfpy 1.4.1 opencensus 0.11.4 opencensus-context 0.1.3 openpyxl 3.1.2 opt-einsum 3.3.0 optax 0.1.9 orbax-checkpoint 0.4.4 overrides 7.7.0 packaging 23.2 pandas 2.2.0 pandocfilters 1.5.1 papermill 2.5.0 parameterized 0.9.0 parso 0.8.3 partd 1.4.1 patsy 0.5.6 pexpect 4.9.0 pillow 10.2.0 pip 20.2.4 platformdirs 4.2.0 pluggy 1.4.0 prometheus-client 0.19.0 prompt-toolkit 3.0.43 protobuf 4.25.2 psutil 5.9.8 ptyprocess 0.7.0 pure-eval 0.2.2 py-cpuinfo 9.0.0 py-spy 0.3.14 py4j 0.10.9.7 pyarrow 15.0.0 pyarrow-hotfix 0.6 pyasn1 0.5.1 pyasn1-modules 0.3.0 pycodestyle 2.11.1 pycparser 2.21 pydantic 2.6.1 pydantic-core 2.16.2 pydot 2.0.0 pyflakes 3.2.0 pygame 2.5.2 pygments 2.17.2 PyOpenGL 3.1.7 pyparsing 3.1.1 Pyro4 4.82 pytest 8.0.0 python-dateutil 2.8.2 python-dotenv 1.0.1 python-json-logger 2.0.7 pytz 2024.1 PyYAML 6.0.1 pyzmq 25.1.2 ray 2.9.2 referencing 0.33.0 regex 2023.12.25 requests 2.31.0 requests-oauthlib 1.3.1 resampy 0.4.2 responses 0.18.0 rfc3339-validator 0.1.4 rfc3986-validator 0.1.1 rich 13.7.0 rpds-py 0.17.1 rsa 4.9 sacremoses 0.1.1 scikit-image 0.22.0 scikit-learn 1.4.0 scipy 1.12.0 seaborn 0.13.2 Send2Trash 1.8.2 serpent 1.41 setuptools 50.3.2 Shimmy 0.2.1 six 1.16.0 smart-open 6.4.0 sniffio 1.3.0 sortedcontainers 2.4.0 soundfile 0.12.1 soupsieve 2.5 stack-data 0.6.3 starlette 0.32.0.post1 statsmodels 0.14.1 swig 4.2.0 symjax 0.5.0 sympy 1.12 tables 3.9.2 tabulate 0.9.0 tblib 3.0.0 tenacity 8.2.3 tensorboard 2.15.2 tensorboard-data-server 0.7.2 tensorboardX 2.6.2.2 tensorflow 2.15.0.post1 tensorflow-estimator 2.15.0 tensorflow-io-gcs-filesystem 0.36.0 tensorflow-probability 0.23.0 tensorly 0.8.1 tensorstore 0.1.45 termcolor 2.4.0 terminado 0.18.0 threadpoolctl 3.2.0 tifffile 2024.2.12 tinycss2 1.2.1 tokenizers 0.12.1 tomli 2.0.1 toolz 0.12.1 torch 2.2.0+cpu torchaudio 2.2.0+cpu torchvision 0.17.0+cpu tornado 6.4 tqdm 4.66.2 traitlets 5.14.1 transformers 4.18.0 typeguard 2.13.3 types-python-dateutil 2.8.19.20240106 typing-extensions 4.9.0 tzdata 2024.1 uri-template 1.3.0 urllib3 2.2.0 uvicorn 0.27.1 uvloop 0.19.0 virtualenv 20.25.0 watchfiles 0.21.0 wcwidth 0.2.13 webcolors 1.13 webencodings 0.5.1 websocket-client 1.7.0 websockets 12.0 werkzeug 3.0.1 wheel 0.42.0 wrapt 1.14.1 xlrd 2.0.1 xlwt 1.3.0 xxhash 3.4.1 yarl 1.9.4 zict 3.0.0 zipp 3.17.0