requirements.txt

# This file may be used to create an environment using:
# $ conda create --name <env> --file <this file>
# platform: linux-64
_libgcc_mutex=0.1=main
_openmp_mutex=5.1=1_gnu
absl-py=1.4.0=pypi_0
antlr4-python3-runtime=4.9.3=pypi_0
anyio=3.7.1=pyhd8ed1ab_0
appdirs=1.4.4=pypi_0
argon2-cffi=23.1.0=pyhd8ed1ab_0
argon2-cffi-bindings=21.2.0=py39hb9d737c_2
ase=3.22.1=pypi_0
asttokens=2.0.5=pyhd3eb1b0_0
attrs=23.2.0=pyh71513ae_0
backcall=0.2.0=pyhd3eb1b0_0
beautifulsoup4=4.12.3=pyha770c72_0
blas=1.0=mkl
bleach=6.1.0=pyhd8ed1ab_0
bottleneck=1.3.5=py39h7deecbd_0
brotlipy=0.7.0=py39h27cfd23_1003
bzip2=1.0.8=h7b6447c_0
ca-certificates=2024.7.4=hbcca054_0
cachetools=5.3.0=pypi_0
certifi=2024.6.2=pyhd8ed1ab_0
cffi=1.15.1=py39h5eee18b_3
charset-normalizer=2.0.4=pyhd3eb1b0_0
click=8.1.3=pypi_0
colorlog=6.7.0=pypi_0
comm=0.1.2=py39h06a4308_0
contourpy=1.0.7=pypi_0
cryptography=39.0.1=py39h9ce1e76_0
cuda=11.6.1=0
cuda-cccl=11.6.55=hf6102b2_0
cuda-command-line-tools=11.6.2=0
cuda-compiler=11.6.2=0
cuda-cudart=12.1.105=0
cuda-cudart-dev=11.6.55=h42ad0f4_0
cuda-cuobjdump=11.6.124=h2eeebcb_0
cuda-cupti=12.1.105=0
cuda-cuxxfilt=11.6.124=hecbf4f6_0
cuda-driver-dev=11.6.55=0
cuda-gdb=12.1.55=0
cuda-libraries=12.1.0=0
cuda-libraries-dev=11.6.1=0
cuda-memcheck=11.8.86=0
cuda-nsight=12.1.55=0
cuda-nsight-compute=12.1.0=0
cuda-nvcc=11.6.124=hbba6d2d_0
cuda-nvdisasm=12.1.55=0
cuda-nvml-dev=11.6.55=haa9ef22_0
cuda-nvprof=12.1.55=0
cuda-nvprune=11.6.124=he22ec0a_0
cuda-nvrtc=12.1.105=0
cuda-nvrtc-dev=11.6.124=h249d397_0
cuda-nvtx=12.1.105=0
cuda-nvvp=12.1.55=0
cuda-opencl=12.4.127=0
cuda-runtime=12.1.0=0
cuda-samples=11.6.101=h8efea70_0
cuda-sanitizer-api=12.1.55=0
cuda-toolkit=11.6.1=0
cuda-tools=11.6.1=0
cuda-visual-tools=11.6.1=0
cycler=0.11.0=pypi_0
cython=0.29.34=pypi_0
data=0.4=pypi_0
debugpy=1.5.1=py39h295c915_0
decorator=4.4.2=py_0
defusedxml=0.7.1=pyhd8ed1ab_0
dgt=1.0.0=dev_0
dill=0.3.6=pypi_0
docker-pycreds=0.4.0=pypi_0
einops=0.7.0=pypi_0
entrypoints=0.4=py39h06a4308_0
exceptiongroup=1.2.0=pyhd8ed1ab_2
executing=0.8.3=pyhd3eb1b0_0
ffmpeg=4.3=hf484d3e_0
filelock=3.13.1=py39h06a4308_0
flake8=6.0.0=pypi_0
flash-attn=2.5.6=pypi_0
flit-core=3.6.0=pyhd3eb1b0_0
fonttools=4.39.3=pypi_0
freetype=2.12.1=h4a9f257_0
fsspec=2024.3.1=pypi_0
funcsigs=1.0.2=pypi_0
future=0.18.3=pypi_0
gds-tools=1.6.0.25=0
giflib=5.2.1=h5eee18b_3
gitdb=4.0.10=pypi_0
gitpython=3.1.31=pypi_0
gmp=6.2.1=h295c915_3
gmpy2=2.1.2=py39heeb90bb_0
gnutls=3.6.15=he1e5248_0
google-auth=2.17.3=pypi_0
google-auth-oauthlib=1.0.0=pypi_0
grpcio=1.54.0=pypi_0
hydra-colorlog=1.2.0=pypi_0
hydra-core=1.3.2=pypi_0
idna=3.4=py39h06a4308_0
importlib-metadata=6.6.0=pypi_0
importlib-resources=5.12.0=pypi_0
importlib_resources=6.4.0=pyhd8ed1ab_0
intel-openmp=2021.4.0=h06a4308_3561
ipdb=0.13.13=pypi_0
ipykernel=6.19.2=py39hb070fc8_0
ipython=8.10.0=py39h06a4308_0
ipython_genutils=0.2.0=pyhd8ed1ab_1
jedi=0.18.1=py39h06a4308_1
jinja2=3.1.3=py39h06a4308_0
joblib=1.2.0=pypi_0
jpeg=9e=h5eee18b_1
jsonschema=4.22.0=pyhd8ed1ab_0
jsonschema-specifications=2023.12.1=pyhd8ed1ab_0
jupyter_client=7.0.6=pyhd8ed1ab_0
jupyter_core=5.2.0=py39h06a4308_0
jupyter_server=1.24.0=pyhd8ed1ab_0
jupyterlab_pygments=0.3.0=pyhd8ed1ab_1
kiwisolver=1.4.5=pypi_0
lame=3.100=h7b6447c_0
latex=0.7.0=pypi_0
lcms2=2.12=h3be6417_0
ld_impl_linux-64=2.38=h1181459_1
lerc=3.0=h295c915_0
libcublas=12.1.0.26=0
libcublas-dev=11.9.2.110=h5c901ab_0
libcufft=11.0.2.4=0
libcufft-dev=10.7.1.112=ha5ce4c0_0
libcufile=1.6.0.25=0
libcufile-dev=1.6.0.25=0
libcurand=10.3.2.56=0
libcurand-dev=10.3.2.56=0
libcusolver=11.4.4.55=0
libcusparse=12.0.2.55=0
libcusparse-dev=11.7.2.124=hbbe9722_0
libdeflate=1.17=h5eee18b_0
libffi=3.4.2=h6a678d5_6
libgcc-ng=11.2.0=h1234567_1
libgomp=11.2.0=h1234567_1
libiconv=1.16=h7f8727e_2
libidn2=2.3.2=h7f8727e_0
libjpeg-turbo=2.0.0=h9bf148f_0
libnpp=12.0.2.50=0
libnpp-dev=11.6.3.124=h3c42840_0
libnvjitlink=12.1.105=0
libnvjpeg=12.1.1.14=0
libnvjpeg-dev=11.6.2.124=hb5906b9_0
libpng=1.6.39=h5eee18b_0
libprotobuf=3.20.3=he621ea3_0
libsodium=1.0.18=h7b6447c_0
libstdcxx-ng=11.2.0=h1234567_1
libtasn1=4.16.0=h27cfd23_0
libtiff=4.5.0=h6a678d5_2
libunistring=0.9.10=h27cfd23_0
libwebp=1.2.4=h11a3e52_1
libwebp-base=1.2.4=h5eee18b_1
llvm-openmp=14.0.6=h9e868ea_0
llvmlite=0.41.1=pypi_0
lz4-c=1.9.4=h6a678d5_0
markdown=3.4.3=pypi_0
markupsafe=2.1.5=pypi_0
matplotlib=3.8.4=pypi_0
matplotlib-inline=0.1.6=py39h06a4308_0
mccabe=0.7.0=pypi_0
mistune=3.0.2=pyhd8ed1ab_0
mkl=2021.4.0=h06a4308_640
mkl-service=2.4.0=py39h7f8727e_0
mkl_fft=1.3.1=py39hd3c417c_0
mkl_random=1.2.2=py39h51133e4_0
mpc=1.1.0=h10f8cd9_1
mpfr=4.0.2=hb69a4c5_1
mpmath=1.3.0=py39h06a4308_0
multinetx=2.3=pypi_0
mypy-extensions=1.0.0=pypi_0
nbclassic=1.1.0=pyhd8ed1ab_0
nbclient=0.10.0=pyhd8ed1ab_0
nbconvert-core=7.16.4=pyhd8ed1ab_1
nbformat=5.10.4=pyhd8ed1ab_0
ncurses=6.4=h6a678d5_0
nest-asyncio=1.5.6=py39h06a4308_0
nettle=3.7.3=hbbd107a_1
networkx=3.2=pypi_0
ninja=1.11.1.1=pypi_0
ninja-base=1.10.2=hd09550d_5
notebook=6.5.7=pyha770c72_0
notebook-shim=0.2.4=pyhd8ed1ab_0
nsight-compute=2023.1.0.15=0
numba=0.58.1=pypi_0
numexpr=2.8.4=py39he184ba9_0
numpy=1.26.4=pypi_0
numpy-base=1.24.3=py39h31eccc5_0
nvidia-cublas-cu12=12.1.3.1=pypi_0
nvidia-cuda-cupti-cu12=12.1.105=pypi_0
nvidia-cuda-nvrtc-cu12=12.1.105=pypi_0
nvidia-cuda-runtime-cu12=12.1.105=pypi_0
nvidia-cudnn-cu12=8.9.2.26=pypi_0
nvidia-cufft-cu12=11.0.2.54=pypi_0
nvidia-curand-cu12=10.3.2.106=pypi_0
nvidia-cusolver-cu12=11.4.5.107=pypi_0
nvidia-cusparse-cu12=12.1.0.106=pypi_0
nvidia-nccl-cu12=2.19.3=pypi_0
nvidia-nvjitlink-cu12=12.4.99=pypi_0
nvidia-nvtx-cu12=12.1.105=pypi_0
oauthlib=3.2.2=pypi_0
omegaconf=2.3.0=pypi_0
openh264=2.1.1=h4ff587b_0
openssl=1.1.1w=h7f8727e_0
packaging=22.0=py39h06a4308_0
pandas=2.2.2=pypi_0
pandocfilters=1.5.0=pyhd8ed1ab_0
parso=0.8.3=pyhd3eb1b0_0
pathtools=0.1.2=pypi_0
pexpect=4.8.0=pyhd3eb1b0_3
pickleshare=0.7.5=pyhd3eb1b0_1003
pillow=10.2.0=pypi_0
pip=22.3.1=py39h06a4308_0
pkgutil-resolve-name=1.3.10=pyhd8ed1ab_1
platformdirs=2.5.2=py39h06a4308_0
prettytable=3.7.0=pypi_0
prometheus_client=0.20.0=pyhd8ed1ab_0
prompt-toolkit=3.0.36=py39h06a4308_0
protobuf=4.22.3=pypi_0
psutil=5.9.0=py39h5eee18b_0
ptyprocess=0.7.0=pyhd3eb1b0_2
pure_eval=0.2.2=pyhd3eb1b0_0
pyasn1=0.5.0=pypi_0
pyasn1-modules=0.3.0=pypi_0
pycodestyle=2.10.0=pypi_0
pycparser=2.21=pyhd3eb1b0_0
pyflakes=3.0.1=pypi_0
pyg-lib=0.4.0+pt22cu121=pypi_0
pygments=2.11.2=pyhd3eb1b0_0
pyopenssl=23.0.0=py39h06a4308_0
pyparsing=2.4.2=pypi_0
pyre-extensions=0.0.30=pypi_0
pysocks=1.7.1=py39h06a4308_0
python=3.9.16=h7a1cb2a_1
python-dateutil=2.8.2=pyhd3eb1b0_0
python-fastjsonschema=2.20.0=pyhd8ed1ab_0
python_abi=3.9=2_cp39
pytorch=2.2.2=py3.9_cuda12.1_cudnn8.9.2_0
pytorch-cuda=12.1=ha16c6d3_5
pytorch-fast-transformers=0.4.0=pypi_0
pytorch-mutex=1.0=cuda
pytz=2022.7=py39h06a4308_0
pyyaml=6.0=py39hb9d737c_4
pyzmq=19.0.2=py39hb69f2a1_2
readline=8.2=h5eee18b_0
referencing=0.35.1=pyhd8ed1ab_0
requests=2.28.1=py39h06a4308_0
requests-oauthlib=1.3.1=pypi_0
rpds-py=0.10.6=py39hb02cf49_0
rsa=4.9=pypi_0
scikit-learn=1.2.2=pypi_0
scikit-network=0.31.0=pypi_0
scipy=1.12.0=pypi_0
seaborn=0.13.2=pypi_0
send2trash=1.8.3=pyh0d859eb_0
sentry-sdk=1.20.0=pypi_0
setproctitle=1.3.2=pypi_0
setuptools=65.6.3=py39h06a4308_0
shutilwhich=1.1.0=pypi_0
six=1.12.0=pypi_0
smmap=5.0.0=pypi_0
sniffio=1.3.1=pyhd8ed1ab_0
soupsieve=2.5=pyhd8ed1ab_1
sqlite=3.40.1=h5082296_0
stack_data=0.2.0=pyhd3eb1b0_0
sympy=1.12=py39h06a4308_0
tempdir=0.7.1=pypi_0
tensorboard=2.12.2=pypi_0
tensorboard-data-server=0.7.0=pypi_0
tensorboard-plugin-wit=1.8.1=pypi_0
terminado=0.18.1=pyh0d859eb_0
threadpoolctl=3.1.0=pypi_0
tinycss2=1.3.0=pyhd8ed1ab_0
tk=8.6.12=h1ccaba5_0
tomli=2.0.1=pypi_0
torch=2.2.1=pypi_0
torch-cluster=1.6.3+pt22cu121=pypi_0
torch-geometric=2.2.0=pypi_0
torch-geometric-temporal=0.54.0=pypi_0
torch-scatter=2.1.2+pt22cu121=pypi_0
torch-sparse=0.6.18+pt22cu121=pypi_0
torch-spline-conv=1.2.2+pt22cu121=pypi_0
torchaudio=2.2.1=pypi_0
torcheval=0.0.6=pypi_0
torchmetrics=0.11.4=pyhd8ed1ab_0
torchtnt=0.0.7=pypi_0
torchtriton=2.2.0=py39
torchvision=0.17.1=pypi_0
tornado=6.2=py39h5eee18b_0
tqdm=4.65.0=pypi_0
traitlets=5.7.1=py39h06a4308_0
triton=2.2.0=pypi_0
tw-benchmark=1.0.0=dev_0
typing-extensions=4.10.0=pypi_0
typing-inspect=0.8.0=pypi_0
typing_extensions=4.11.0=pyha770c72_0
tzdata=2024.1=pypi_0
urllib3=1.26.14=py39h06a4308_0
wandb=0.15.2=pypi_0
wcwidth=0.2.5=pyhd3eb1b0_0
webencodings=0.5.1=pyhd8ed1ab_2
websocket-client=1.8.0=pyhd8ed1ab_0
werkzeug=2.2.3=pypi_0
wheel=0.38.4=py39h06a4308_0
xz=5.2.10=h5eee18b_1
yaml=0.2.5=h7f98852_2
zeromq=4.3.4=h2531618_0
zipp=3.15.0=pypi_0
zlib=1.2.13=h5eee18b_0
zstd=1.5.2=ha4553b6_0