--------------------- Module Specific Help for "jax/0.5.0" --------------------- JAX is Autograd and XLA, brought together for high-performance machine learning research. JAX 0.5.0 library for Python 3.12.9 and CUDA 12.2 This module provides JAX with CUDA support, as well as tensorflow and pytorch (both CPU only) for utility functions and a large number of popular JAX-based machine learning packages installed. If you want more control over installed packages, the "small" jax module comes with only JAX, tensorflow and pytorch. 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 aiohappyeyeballs 2.4.6 aiohttp 3.11.12 aiohttp-cors 0.7.0 aiosignal 1.3.2 annotated-types 0.7.0 ansicolors 1.1.8 anyio 4.8.0 argon2-cffi 23.1.0 argon2-cffi-bindings 21.2.0 arrow 1.3.0 asttokens 3.0.0 astunparse 1.6.3 async-lru 2.0.4 attrs 25.1.0 babel 2.17.0 bayesian-optimization 2.0.3 beautifulsoup4 4.13.3 bleach 6.2.0 blosc2 3.0.0 cachetools 5.5.1 certifi 2025.1.31 cffi 1.17.1 cftime 1.6.4.post1 charset-normalizer 3.4.1 chex 0.1.88 click 8.1.8 cloudpickle 3.1.1 colorama 0.4.6 colorful 0.5.6 comm 0.2.2 ConfigSpace 1.2.1 contourpy 1.3.1 cycler 0.12.1 Cython 3.0.12 dask 2025.1.0 dask-jobqueue 0.9.0 dataclasses 0.6 datasets 3.2.0 debugpy 1.8.12 decorator 5.1.1 defusedxml 0.7.1 dill 0.3.8 distlib 0.3.9 distrax 0.1.5 distributed 2025.1.0 dm-haiku 0.0.13 dm-tree 0.1.9 e3nn-jax 0.20.7 einops 0.8.1 entrypoints 0.4 equinox 0.11.11 et_xmlfile 2.0.0 etils 1.12.0 evaluate 0.4.3 executing 2.2.0 fastapi 0.115.8 fastjsonschema 2.21.1 filelock 3.13.1 flatbuffers 25.2.10 flax 0.10.3 fonttools 4.56.0 fqdn 1.5.1 frozenlist 1.5.0 fsspec 2024.6.1 future 1.0.0 gast 0.6.0 google-api-core 2.24.1 google-auth 2.38.0 google-pasta 0.2.0 googleapis-common-protos 1.67.0rc1 graphviz 0.20.3 grpcio 1.70.0 h11 0.14.0 h5py 3.12.1 hpbandster 0.7.4 httpcore 1.0.7 httptools 0.6.4 httpx 0.28.1 huggingface-hub 0.28.1 humanize 4.11.0 hyperopt 0.2.7 idna 3.10 imageio 2.37.0 imbalanced-learn 0.13.0 importlib_resources 6.5.2 iniconfig 2.0.0 ipykernel 6.29.5 ipython 8.32.0 ipython-genutils 0.2.0 ipywidgets 8.1.5 isoduration 20.11.0 jax 0.5.0 jax-cuda12-pjrt 0.5.0 jax-cuda12-plugin 0.5.0 jax-md 0.2.8 jaxlib 0.5.0 jaxtyping 0.2.37 jedi 0.19.2 Jinja2 3.1.4 jmp 0.0.4 joblib 1.4.2 jraph 0.0.6.dev0 json5 0.10.0 jsonpointer 3.0.0 jsonschema 4.23.0 jsonschema-specifications 2024.10.1 jupyter 1.1.1 jupyter_client 8.6.3 jupyter-console 6.6.3 jupyter_core 5.7.2 jupyter-events 0.12.0 jupyter-lsp 2.2.5 jupyter_server 2.15.0 jupyter_server_terminals 0.5.3 jupyterlab 4.3.5 jupyterlab_pygments 0.3.0 jupyterlab_server 2.27.3 jupyterlab_widgets 3.0.13 keras 3.8.0 kiwisolver 1.4.8 lazy_loader 0.4 libclang 18.1.1 llvmlite 0.44.0 lmdb 1.6.2 locket 1.0.0 lxml 5.3.1 Markdown 3.7 markdown-it-py 3.0.0 MarkupSafe 2.1.5 matplotlib 3.10.0 matplotlib-inline 0.1.7 mdurl 0.1.2 mistune 3.1.1 ml_collections 1.0.0 ml-dtypes 0.4.1 more-itertools 10.6.0 mpmath 1.3.0 msgpack 1.1.0 multidict 6.1.0 multiprocess 0.70.16 mysql-connector-python 9.2.0 namex 0.0.8 nbclassic 1.2.0 nbclient 0.10.2 nbconvert 7.16.6 nbformat 5.10.4 ndindex 1.9.2 nest-asyncio 1.6.0 netCDF4 1.7.2 netifaces 0.11.0 networkx 3.3 notebook 7.3.2 notebook_shim 0.2.4 numba 0.61.0 numexpr 2.10.2 numpy 2.0.2 objax 1.8.0 odfpy 1.4.1 opencensus 0.11.4 opencensus-context 0.1.3 openpyxl 3.1.5 opt_einsum 3.4.0 optax 0.2.4 optree 0.14.0 orbax-checkpoint 0.11.5 overrides 7.7.0 packaging 24.2 pandas 2.2.3 pandocfilters 1.5.1 papermill 2.6.0 parameterized 0.9.0 parso 0.8.4 partd 1.4.2 patsy 1.0.1 pexpect 4.9.0 pillow 11.0.0 pip 25.0.1 platformdirs 4.3.6 pluggy 1.5.0 prometheus_client 0.21.1 prompt_toolkit 3.0.50 propcache 0.2.1 proto-plus 1.26.0 protobuf 5.29.3 psutil 6.1.1 ptyprocess 0.7.0 pure_eval 0.2.3 py-cpuinfo 9.0.0 py-spy 0.4.0 py4j 0.10.9.9 pyarrow 19.0.0 pyasn1 0.6.1 pyasn1_modules 0.4.1 pycparser 2.22 pydantic 2.10.6 pydantic_core 2.27.2 pydot 3.0.4 pyflakes 3.2.0 Pygments 2.19.1 pyparsing 3.2.1 Pyro4 4.82 pytest 8.3.4 python-dateutil 2.9.0.post0 python-dotenv 1.0.1 python-json-logger 3.2.1 pytz 2025.1 PyYAML 6.0.2 pyzmq 26.2.1 ray 2.42.1 referencing 0.36.2 regex 2024.11.6 requests 2.32.3 resampy 0.4.3 rfc3339-validator 0.1.4 rfc3986-validator 0.1.1 rich 13.9.4 rpds-py 0.22.3 rsa 4.9 scikit-image 0.25.1 scikit-learn 1.6.1 scipy 1.15.1 seaborn 0.13.2 Send2Trash 1.8.3 serpent 1.41 setuptools 70.2.0 simplejson 3.19.3 six 1.17.0 sklearn-compat 0.1.3 smart-open 7.1.0 sniffio 1.3.1 sortedcontainers 2.4.0 soundfile 0.13.1 soupsieve 2.6 stack-data 0.6.3 starlette 0.45.3 statsmodels 0.14.4 symjax 0.5.0 sympy 1.13.1 tables 3.10.2 tabulate 0.9.0 tblib 3.0.0 tenacity 9.0.0 tensorboard 2.18.0 tensorboard-data-server 0.7.2 tensorboardX 2.6.2.2 tensorflow 2.18.0 tensorflow_cpu 2.18.0 tensorflow-probability 0.25.0 tensorly 0.9.0 tensorstore 0.1.71 termcolor 2.5.0 terminado 0.18.1 tf_keras 2.18.0 threadpoolctl 3.5.0 tifffile 2025.1.10 tinycss2 1.4.0 toolz 1.0.0 torch 2.6.0+cpu torchaudio 2.6.0+cpu torchvision 0.21.0+cpu tornado 6.4.2 tqdm 4.67.1 traitlets 5.14.3 treescope 0.1.8 types-python-dateutil 2.9.0.20241206 typing_extensions 4.12.2 tzdata 2025.1 uri-template 1.3.0 urllib3 2.3.0 uvicorn 0.34.0 uvloop 0.21.0 virtualenv 20.29.2 wadler_lindig 0.1.3 watchfiles 1.0.4 wcwidth 0.2.13 webcolors 24.11.1 webencodings 0.5.1 websocket-client 1.8.0 websockets 14.2 Werkzeug 3.1.3 wheel 0.45.1 widgetsnbextension 4.0.13 wrapt 1.17.2 xlrd 2.0.1 xlwt 1.3.0 xxhash 3.5.0 yarl 1.18.3 zict 3.0.0 zipp 3.21.0