# Environment record — the versions this project's published results were produced on.
#
# This is NOT a `pip install -r` recipe, and it is deliberately not wired into the
# packaging. Two reasons:
#
#   1. 76 of the entries below are `@ file://` wheels baked into the NGC image --
#      torch, torchvision and torch_tensorrt among them. They do not exist on PyPI,
#      so this file cannot recreate the environment; the container image can.
#   2. This is the whole container, including packages micm-nlp never imports.
#   3. micm-nlp itself is omitted. `pip freeze` renders an editable install as its
#      git URL, and this repo's remote carries credentials -- so that line would
#      have committed a token. Never paste `pip freeze` output in unread.
#
# What it is for: saying precisely what ran, when a number in a paper needs an
# environment attached to it. `pyproject.toml` says what *can* work; this says what
# *was* run.
#
# Captured from the project container
# (/netscratch/bmikaberidze/micm_nlp.nvcr.io_nvidia_pytorch_24.08-py3.sqsh) on
# 2026-09-04. Regenerate with:
#
#   sbatch --array=0 --mem=30G runtime/clusters/pegasus/shell/run.sh \
#     --site-packages --no-gpu "python -m pip freeze"
#
absl-py==2.1.0
accelerate==1.3.0
aenum==3.1.17
aiohappyeyeballs @ file:///rapids/aiohappyeyeballs-2.3.5-py3-none-any.whl#sha256=4d6dea59215537dbc746e93e779caea8178c866856a721c9c660d7a5a7b8be03
aiohttp @ file:///rapids/aiohttp-3.10.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=b9db600a86414a9a653e3c1c7f6a2f6a1894ab8f83d11505247bd1b90ad57157
aiosignal @ file:///rapids/aiosignal-1.3.1-py3-none-any.whl#sha256=f8376fb07dd1e86a584e4fcdec80b36b7f81aac666ebc724e2c090300dd83b17
annotated-types==0.7.0
anyio==4.4.0
apex @ file:///opt/pytorch/apex
argon2-cffi==23.1.0
argon2-cffi-bindings==21.2.0
arrow==1.3.0
asciitree @ file:///rapids/asciitree-0.3.3-py3-none-any.whl#sha256=f244fc3c06b0e9a492fecede9f4d9bd8b400b192cfe610c061a1bc918e16d8b5
asttokens==2.4.1
astunparse==1.6.3
async-lru==2.0.4
async-timeout @ file:///rapids/async_timeout-4.0.3-py3-none-any.whl#sha256=7405140ff1230c310e51dc27b3145b9092d659ce68ff733fb0cefe3ee42be028
attrs==24.2.0
audioread==3.0.1
babel==2.16.0
beautifulsoup4==4.12.3
better-exceptions==0.3.3
black==24.8.0
bleach==6.1.0
blis==0.7.11
cachetools @ file:///rapids/cachetools-5.4.0-py3-none-any.whl#sha256=3ae3b49a3d5e28a77a0be2b37dbcb89005058959cb2323858c2657c4a8cab474
catalogue==2.0.10
certifi==2024.7.4
cffi==1.17.0
charset-normalizer==3.3.2
click==8.1.7
cloudpathlib==0.18.1
cloudpickle @ file:///rapids/cloudpickle-3.0.0-py3-none-any.whl#sha256=246ee7d0c295602a036e86369c77fecda4ab17b506496730f2f576d9016fd9c7
cmake==3.30.2
comm==0.2.2
confection==0.1.5
contourpy==1.2.1
cuda-python @ file:///rapids/cuda_python-12.5.0-cp310-cp310-linux_x86_64.whl#sha256=81b0ba290576c028ecfed9ac101c50a6239e24c515a8ac8f506ccf8184dae126
cudf @ file:///rapids/cudf-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=7624c0ef3d7bd9825763824900dd516bb82262837d71589ed52fc0bd4c81e4c0
cugraph @ file:///rapids/cugraph-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=a67228b4dcd11a1f34c8d7e46359d251dd73b370ff1204d7b47d414abb3cbc3e
cugraph-dgl @ file:///rapids/cugraph_dgl-24.6.0-py3-none-any.whl#sha256=24667bde351e4444e251d988903c54142dee989ce2fa4ad8eb8ed05ea7231be4
cugraph-equivariant @ file:///rapids/cugraph_equivariant-24.6.0-py3-none-any.whl#sha256=f2a95f03dbda96c499a14a90f14da334e0c9ad066902166491c84198c2203bcd
cugraph-pyg @ file:///rapids/cugraph_pyg-24.6.0-py3-none-any.whl#sha256=03beb173322f8f0685e6024ffdb8e401b589c5078a3a8be7e7a075835584ce51
cugraph-service-client @ file:///rapids/cugraph_service_client-24.6.0-py3-none-any.whl#sha256=ed5e24b8113a33b941632b17c129cccf77a16534cd4a4980549a3fb98a6601a5
cugraph-service-server @ file:///rapids/cugraph_service_server-24.6.0-py3-none-any.whl#sha256=2392a7814834e9596bb6d2639c439227d205d7f7792fe58afd5a3b3362e8b4b8
cuml @ file:///rapids/cuml-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=805fa843ef6385fdaf731c51ef2044a951db2128bce9505f4dc8ee3962c1c2be
cupy-cuda12x @ file:///rapids/cupy_cuda12x-13.2.0-cp310-cp310-linux_x86_64.whl#sha256=fc84944ab27723e44024ccc551b27fad6cdbc6936ad74867c933c129c16fd170
cycler==0.12.1
cymem==2.0.8
Cython==3.0.11
dask @ file:///rapids/dask-2024.5.1-py3-none-any.whl#sha256=af1cadd1fd1d1d44600ff5de43dd029e5668fdf87422131f4e3e3aa2a6a63555
dask-cuda @ file:///rapids/dask_cuda-24.6.0-py3-none-any.whl#sha256=76f15e7f3afe5778e5d1ade7cc4384bb89938fbf18b25f7aad891353ccbb66b9
dask-cudf @ file:///rapids/dask_cudf-24.6.0-py3-none-any.whl#sha256=477310f987a69949ef9077d3a71277715d38e6398c24bd58dd58af0d745e4867
dask-expr @ file:///rapids/dask_expr-1.1.1-py3-none-any.whl#sha256=455842bff0795309f75f78f98d9be235210e01ad93da9b5a74f8cdaa594ab958
datasets==3.2.0
datasketch==1.6.5
debugpy==1.8.5
decorator==5.1.1
defusedxml==0.7.1
dill==0.3.8
distributed @ file:///rapids/distributed-2024.5.1-py3-none-any.whl#sha256=cf77583ce67f6696aa716c30f1cba3403ac866241c00d6e9a0bc1500c6818b40
distributed-ucxx @ file:///rapids/distributed_ucxx-0.38.0-py3-none-any.whl#sha256=2c5bc282c2a382c268342e884f887f49a019dca5d850cf6b2609fd7856f812d3
dm-tree==0.1.8
docker-pycreds==0.4.0
einops==0.8.0
entrypoints @ file:///rapids/entrypoints-0.4-py3-none-any.whl#sha256=f174b5ff827504fd3cd97cc3f8649f3693f51538c7e4bdf3ef002c8429d42f9f
evaluate==0.4.3
exceptiongroup==1.2.2
execnet==2.1.1
executing==2.0.1
expecttest==0.1.3
fasteners @ file:///rapids/fasteners-0.19-py3-none-any.whl#sha256=758819cb5d94cdedf4e836988b74de396ceacb8e2794d21f82d131fd9ee77237
fastjsonschema==2.20.0
fastrlock @ file:///rapids/fastrlock-0.8.2-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_28_x86_64.whl#sha256=08315bde19d0c2e6b06593d5a418be3dc8f9b1ee721afa96867b9853fceb45cf
fasttext==0.9.3
filelock==3.15.4
flash_attn==2.4.2
fonttools==4.53.1
fqdn==1.5.1
frozenlist @ file:///rapids/frozenlist-1.4.1-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=a9b2de4cf0cdd5bd2dee4c4f63a653c61d2408055ab77b151c1957f221cabf2a
fsspec==2024.6.1
gast==0.6.0
gitdb==4.0.12
GitPython==3.1.44
grpcio @ file:///rapids/grpcio-1.62.1-cp310-cp310-linux_x86_64.whl#sha256=f7cce74ba70c9dd9f0ee13e4c7d51fc875dffc5182254af8b3c1ba317ea87822
h11==0.14.0
httpcore==1.0.5
httpx==0.27.0
huggingface-hub==0.28.1
hypothesis==5.35.1
idna==3.7
igraph==0.11.6
importlib_metadata @ file:///rapids/importlib_metadata-7.2.1-py3-none-any.whl#sha256=ffef94b0b66046dd8ea2d619b701fe978d9264d38f3998bc4c27ec3b146a87c8
iniconfig==2.0.0
intel-openmp==2021.4.0
ipykernel==6.29.5
ipython==8.26.0
isoduration==20.11.0
isort==5.13.2
jedi==0.19.1
Jinja2==3.1.4
joblib==1.4.2
json5==0.9.25
jsonpointer==3.0.0
jsonschema==4.23.0
jsonschema-specifications==2023.12.1
jupyter-events==0.10.0
jupyter-lsp==2.2.5
jupyter_client==8.6.2
jupyter_core==5.7.2
jupyter_server==2.14.2
jupyter_server_terminals==0.5.3
jupyterlab==4.2.4
jupyterlab-tensorboard-pro==4.0.0
jupyterlab_code_formatter==3.0.2
jupyterlab_pygments==0.3.0
jupyterlab_server==2.27.3
jupytext==1.16.4
kiwisolver==1.4.5
kvikio @ file:///rapids/kvikio-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=2d63bc31c957748fd34cb5e14ca4563aa4ad5d5246aba1e69ce1101fe082a3e4
langcodes==3.4.0
language_data==1.2.0
lazy_loader==0.4
librmm @ file:///rapids/librmm-24.6.0-py3-none-linux_x86_64.whl#sha256=7646b7a415491e94991ff48204ecfd283480c1e8187618b219be9d91d066baa1
librosa==0.10.1
lightning==2.5.0.post0
lightning-thunder==0.2.0.dev0
lightning-utilities==0.11.6
lintrunner==0.12.5
llvmlite @ file:///rapids/llvmlite-0.42.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=763f8d8717a9073b9e0246998de89929071d15b47f254c10eef2310b9aac033d
locket @ file:///rapids/locket-1.0.0-py2.py3-none-any.whl#sha256=b6c819a722f7b6bd955b80781788e4a66a55628b858d347536b7e81325a3a5e3
looseversion==1.3.0
marisa-trie==1.2.0
Markdown==3.7
markdown-it-py==3.0.0
MarkupSafe==2.1.5
matplotlib==3.9.2
matplotlib-inline==0.1.7
mdit-py-plugins==0.4.1
mdurl==0.1.2
mistune==3.0.2
mkl==2021.1.1
mkl-devel==2021.1.1
mkl-include==2021.1.1
mock==5.1.0
mpmath==1.3.0
msgpack==1.0.8
multidict @ file:///rapids/multidict-6.0.5-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=21fd81c4ebdb4f214161be351eb5bcf385426bf023041da2fd9e60681f3cebae
multiprocess==0.70.16
murmurhash==1.0.10
mypy-extensions==1.0.0
nbclient==0.10.0
nbconvert==7.16.4
nbformat==5.10.4
nest-asyncio==1.6.0
networkx==3.3
ninja==1.11.1.1
nltk==3.9.1
notebook==7.2.1
notebook_shim==0.2.4
numba @ file:///rapids/numba-0.59.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl#sha256=525ef3f820931bdae95ee5379c670d5c97289c6520726bc6937a4a7d4230ba24
numcodecs @ file:///rapids/numcodecs-0.11.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=c27dfca402f69fbfa01c46fb572086e77f38121192160cc8ed1177dc30702c52
numpy==1.24.4
nvfuser==0.2.6a0+1d02b13
nvidia-cudnn-frontend @ file:///opt/pytorch/lightning-thunder/tmp/cudnn_frontend
nvidia-dali-cuda120==1.40.0
nvidia-ml-py3==7.352.0
nvidia-modelopt==0.15.0
nvidia-nvimgcodec-cu12==0.3.0.5
nvidia-pyindex==1.0.9
nvtx @ file:///rapids/nvtx-0.2.5-cp310-cp310-linux_x86_64.whl#sha256=06b293a97423f769ee026400381e75b1a82bb7a9bdb6ed72087a4e9b5aeea4c1
nx-cugraph @ file:///rapids/nx_cugraph-24.6.0-py3-none-any.whl#sha256=c0c63fb07c5ce71dda8a9c3229de6a63d23f3b538abd18c08454f03bdf07d8cb
onnx @ file:///opt/pytorch/pytorch/third_party/onnx
opencv @ file:///opencv-4.7.0/modules/python/package
openTSNE==1.0.4
opt-einsum==3.3.0
optree==0.12.1
overrides==7.7.0
packaging @ file:///rapids/packaging-23.2-py3-none-any.whl#sha256=8c491190033a9af7e1d931d0b5dacc2ef47509b34dd0de67ed209b5203fc88c7
pandas @ file:///rapids/pandas-2.2.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=8635c16bf3d99040fdf3ca3db669a7250ddf49c55dc4aa8fe0ae0fa8d6dcc1f0
pandocfilters==1.5.1
parso==0.8.4
partd @ file:///rapids/partd-1.4.2-py3-none-any.whl#sha256=978e4ac767ec4ba5b86c6eaa52e5a2a3bc748a2ca839e8cc798f1cc6ce6efb0f
pathspec==0.12.1
peft==0.14.0
pexpect==4.9.0
pillow==10.4.0
platformdirs==4.2.2
pluggy==1.5.0
ply @ file:///rapids/ply-3.11-py2.py3-none-any.whl#sha256=096f9b8350b65ebd2fd1346b12452efe5b9607f7482813ffca50c22722a807ce
polygraphy==0.49.12
pooch==1.8.2
preshed==3.0.9
prometheus_client==0.20.0
prompt_toolkit==3.0.47
protobuf==4.24.4
psutil==6.0.0
ptyprocess==0.7.0
PuLP==2.9.0
pure_eval==0.2.3
pyarrow @ file:///rapids/pyarrow-16.1.0-cp310-cp310-linux_x86_64.whl#sha256=4f519cb9d9f4e90b10fe3dd381f53afd3e89edfc2e3e525fe16d1483cafacb49
pybind11==2.13.4
pybind11_global==2.13.4
pycocotools @ git+https://github.com/nvidia/cocoapi.git@d99cbf3823588ef09a2721655f46e509ebafb3d7#subdirectory=PythonAPI
pycparser==2.22
pydantic==2.8.2
pydantic-settings==2.14.0
pydantic_core==2.20.1
Pygments==2.18.0
pylibcugraph @ file:///rapids/pylibcugraph-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=ce84be6aae4aa7043cbf96a357c65b6d20b5dd8962fa0ee60ce9afc8b919ff14
pylibcugraphops @ file:///rapids/pylibcugraphops-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=bf681e01a9226fee665157dda0762bd03f330b68d581e051ecea5eca7a1fc878
pylibraft @ file:///rapids/pylibraft-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=bed50e392824eee3bd37b5d4f40bd41f86f9037cd89f7dae274838dc692d4876
pylibwholegraph @ file:///rapids/pylibwholegraph-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=48d0ea366ee02cb41858a25b445640605e60cf8fa85b8f908d8537760fff1661
pynvjitlink @ file:///rapids/pynvjitlink-0.2.3-cp310-cp310-linux_x86_64.whl#sha256=108ea553c8246cb8850c2f6ca37d2de6719a8aed61118d64e4d94a1b4fa31734
pynvml @ file:///rapids/pynvml-11.4.1-py3-none-any.whl#sha256=d27be542cd9d06558de18e2deffc8022ccd7355bc7382255d477038e7e424c6c
pyparsing==3.1.2
pypdf==6.15.0
pytest==8.1.1
pytest-flakefinder==1.1.0
pytest-rerunfailures==14.0
pytest-shard==0.1.2
pytest-xdist==3.6.1
python-dateutil==2.9.0.post0
python-dotenv==1.0.1
python-hostlist==1.23.0
python-json-logger==2.0.7
pytorch-lightning==2.5.0.post0
pytorch-triton @ file:///tmp/dist/pytorch_triton-3.0.0%2Bdedb7bdf3-cp310-cp310-linux_x86_64.whl#sha256=d6f770e5e6b195feb8ba9c3ac057ced4b8faee549cbc07c94f4aaeca4443f6e6
pytz @ file:///rapids/pytz-2023.4-py2.py3-none-any.whl#sha256=f90ef520d95e7c46951105338d918664ebfd6f1d995bd7d153127ce90efafa6a
PyYAML==6.0.2
pyzmq==26.1.0
raft-dask @ file:///rapids/raft_dask-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=2405a90ea7f4b18f125e277c21a17e2a1ca3a79d84198e18f304adbe899a96e0
rapids-dask-dependency @ file:///rapids/rapids_dask_dependency-24.6.0a0-py3-none-any.whl#sha256=691747bea035c7433626adc15921859e367dddca953421f8302f3a2a8808fe30
referencing==0.35.1
regex==2024.7.24
requests==2.32.3
rfc3339-validator==0.1.4
rfc3986-validator==0.1.1
rich==13.7.1
rmm @ file:///rapids/rmm-24.6.0-cp310-cp310-linux_x86_64.whl#sha256=bf99f1f6b15deae43e097a2386258ab6da27abc1c4cac3ebe4a2dd4c08243ebd
rpds-py==0.20.0
ruff==0.15.12
safetensors==0.5.2
scikit-learn==1.5.1
scipy @ file:///rapids/scipy-1.14.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=42470ea0195336df319741e230626b6225a740fd9dce9642ca13e98f667047c0
Send2Trash==1.8.3
sentencepiece==0.2.0
sentry-sdk==2.20.0
seqeval==1.2.2
setproctitle==1.3.4
shellingham==1.5.4
six==1.16.0
smart-open==7.0.4
smmap==5.0.2
sniffio==1.3.1
sortedcontainers==2.4.0
soundfile==0.12.1
soupsieve==2.6
soxr==0.4.0
spacy==3.7.5
spacy-legacy==3.0.12
spacy-loggers==1.0.5
srsly==2.4.8
stack-data==0.6.3
sympy==1.13.1
tabulate==0.9.0
tbb==2021.13.1
tblib @ file:///rapids/tblib-3.0.0-py3-none-any.whl#sha256=80a6c77e59b55e83911e1e607c649836a69c103963c5f28a46cbeef44acf8129
tensorboard==2.16.2
tensorboard-data-server==0.7.2
tensorrt @ file:///workspace/TensorRT-10.3.0.26/python/tensorrt-10.3.0-cp310-none-linux_x86_64.whl#sha256=742a9f2a1e95e89e8976ce5754e1994d070dfe74cffd423cf3d36d1de022401e
terminado==0.18.1
texttable==1.7.0
thinc==8.2.5
threadpoolctl==3.5.0
thriftpy2 @ file:///rapids/thriftpy2-0.5.0-cp310-cp310-linux_x86_64.whl#sha256=a7cb6fb5225ea67b744f406d25f271b6827513dac68da1109d70aa03a3d8d8cb
tinycss2==1.3.0
tokenizers==0.21.0
tomli==2.0.1
toolz @ file:///rapids/toolz-0.12.1-py3-none-any.whl#sha256=d22731364c07d72eea0a0ad45bafb2c2937ab6fd38a3507bf55eae8744aa7d85
torch @ file:///opt/transfer/torch-2.5.0a0%2B872d972e41.nv24.8-cp310-cp310-linux_x86_64.whl#sha256=02c7278cb9c4879264677ba898a67a8cb9f31961779d2ac01298489aabec526e
torch_tensorrt @ file:///opt/pytorch/torch_tensorrt/dist/torch_tensorrt-2.5.0a0-cp310-cp310-linux_x86_64.whl#sha256=69fbc675bfbc3ed552d1b7c47d2a24a7c2e29531f8d56fac5c98805133d16e46
torchmetrics==1.6.1
torchvision @ file:///opt/pytorch/vision
tornado==6.2
tqdm==4.66.5
traitlets==5.14.3
transformer_engine @ git+https://github.com/NVIDIA/TransformerEngine.git@ba36f90d05c203787294b7e490af901d79f07d30
transformers==4.48.2
treelite @ file:///rapids/treelite-4.1.2-py3-none-linux_x86_64.whl#sha256=1519a67b00539079c437ee2f9dbce489af2329ab5bd135e9b6f4c937c4ce5cc3
typer==0.12.4
types-dataclasses==0.6.6
types-python-dateutil==2.9.0.20240316
typing-inspection==0.4.2
typing_extensions==4.12.2
tzdata @ file:///rapids/tzdata-2024.1-py2.py3-none-any.whl#sha256=9068bc196136463f5245e51efda838afa15aaeca9903f49050dfa2679db4d252
ucx-py @ file:///rapids/ucx_py-0.38.0-cp310-cp310-linux_x86_64.whl#sha256=c0109ba3b7c192f5709c79e55119e3e8d43bb3e33fe1eba1b7031946483d24ea
ucxx @ file:///rapids/ucxx-0.38.0-cp310-cp310-linux_x86_64.whl#sha256=dee372074c782963238af1346ba075badaa224b1e52130eab0061e89751ad5fe
uri-template==1.3.0
urllib3 @ file:///rapids/urllib3-2.0.7-py3-none-any.whl#sha256=fdb6d215c776278489906c2f8916e6e7d4f5a9b602ccbcfdf7f016fc8da0596e
wandb==0.19.5
wasabi==1.1.3
wcwidth==0.2.13
weasel==0.4.1
webcolors==24.8.0
webencodings==0.5.1
websocket-client==1.8.0
Werkzeug==3.0.3
wrapt==1.16.0
xdoctest==1.0.2
xgboost @ file:///rapids/xgboost-2.0.3-py3-none-linux_x86_64.whl#sha256=eb148d053b70361500ed7f26068f1f961969572f38faf2da980a024d216a8918
xxhash==3.5.0
yarl @ file:///rapids/yarl-1.9.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl#sha256=357495293086c5b6d34ca9616a43d329317feab7917518bc97a08f9e55648455
zarr @ file:///rapids/zarr-2.18.2-py3-none-any.whl#sha256=a638754902f97efa99b406083fdc807a0e2ccf12a949117389d2a4ba9b05df38
zict @ file:///rapids/zict-3.0.0-py2.py3-none-any.whl#sha256=5796e36bd0e0cc8cf0fbc1ace6a68912611c1dbd74750a3f3026b9b9d6a327ae
zipp @ file:///rapids/zipp-3.19.2-py3-none-any.whl#sha256=f091755f667055f2d02b32c53771a7a6c8b47e1fdbc4b72a8b9072b3eef8015c
