absl-py==2.3.1
accelerate==1.9.0
aiohappyeyeballs==2.6.1
aiohttp==3.12.15
aiosignal==1.4.0
-e git+https://github.com/rocm/aiter.git@3d48b72be951d4179f045acd804109ac77ea4534#egg=aiter
amdsmi @ file:///opt/rocm-7.0.0/share/amd_smi
aniso8601==10.0.1
annotated-types==0.7.0
antlr4-python3-runtime==4.9.3
anyio==4.11.0
apex @ file:///apex-1.9.0a0%2Brocm7.0.0.git4b035815-cp310-cp310-linux_x86_64.whl#sha256=34c6e78776bd649e49902fa9fb5c20a044ea134aa608cd594b5042830369ed64
argon2-cffi==25.1.0
argon2-cffi-bindings==25.1.0
arrow==1.3.0
asciitree==0.3.3
asttokens==3.0.0
async-lru==2.0.5
async-timeout==4.0.3
attrs==25.3.0
audioread==3.0.1
babel==2.17.0
beautifulsoup4==4.14.2
bleach==6.2.0
blinker==1.9.0
blobfile==3.1.0
boto3==1.35.42
botocore==1.35.99
build==1.3.0
causal_conv1d @ file:///workspace/causal-conv1d
certifi==2025.8.3
cffi==1.17.1
charset-normalizer==3.4.3
click==8.2.1
cloudpickle==3.1.1
cmake==3.31.6
colorama==0.4.6
comm==0.2.3
contourpy==1.3.2
coremltools==5.0b5
coverage==7.10.6
cryptography==45.0.6
cuda-bindings==12.9.2
cuda-pathfinder==1.1.0
cxxfilt==0.3.0
cycler==0.12.1
dask==2025.9.1
dataclasses-json==0.6.7
datasets==4.0.0
debugpy==1.8.17
decorator==5.2.1
deepmerge==2.0
defusedxml==0.7.1
Deprecated==1.2.18
dill==0.3.8
distributed==2025.9.1
distro==1.9.0
docstring_parser==0.17.0
einops==0.8.1
exceptiongroup==1.3.0
execnet==2.1.1
executing==2.2.1
expecttest==0.3.0
faiss-cpu==1.12.0
fasteners==0.20
fastjsonschema==2.21.2
fbscribelogger==0.1.7
filelock==3.18.0
flash_attn==2.8.3
Flask==3.1.2
Flask-RESTful==0.3.10
flatbuffers==24.12.23
fonttools==4.59.1
fqdn==1.5.1
frozenlist==1.7.0
fsspec==2025.3.0
future==1.0.0
geopandas==1.1.3
ghstack==0.8.0
gitdb==4.0.12
GitPython==3.1.45
greenlet==3.2.4
grouped_gemm @ file:///workspace/grouped_gemm
grpcio==1.74.0
h11==0.16.0
hf-xet==1.1.9
hf_transfer==0.1.9
httpcore==1.0.9
httpx==0.28.1
httpx-sse==0.4.1
huggingface-hub==0.34.4
hypothesis==5.35.1
idna==3.10
ihighlight==1.0.0
imageio==2.37.0
importlib_metadata==8.7.0
iniconfig==2.1.0
inquirerpy==0.3.4
ipykernel==6.30.1
ipython==8.37.0
ipywidgets==8.1.7
isoduration==20.11.0
itsdangerous==2.2.0
jedi==0.19.2
Jinja2==3.1.6
jmespath==1.0.1
joblib==1.5.1
json5==0.12.1
jsonpatch==1.33
jsonpath-ng==1.7.0
jsonpointer==3.0.0
jsonschema==4.25.1
jsonschema-specifications==2025.9.1
junitparser==2.1.1
jupyter-events==0.12.0
jupyter-lsp==2.3.0
jupyter_ai==2.31.4
jupyter_ai_magics==2.31.6
jupyter_client==8.6.3
jupyter_core==5.8.1
jupyter_server==2.17.0
jupyter_server_terminals==0.5.3
jupyterlab==4.4.2
jupyterlab-lsp==5.2.0
jupyterlab_pygments==0.3.0
jupyterlab_server==2.27.3
jupyterlab_widgets==3.0.15
kagglehub==0.3.13
kiwisolver==1.4.9
langchain==0.3.27
langchain-community==0.3.30
langchain-core==0.3.76
langchain-text-splitters==0.3.11
langsmith==0.4.31
lark==1.3.0
lazy_loader==0.4
librosa==0.11.0
lintrunner==0.12.7
llvmlite==0.44.0
locket==1.0.0
lxml==5.3.0
Markdown==3.8.2
markdown-it-py==4.0.0
MarkupSafe==3.0.2
marshmallow==3.26.1
matplotlib==3.10.5
matplotlib-inline==0.1.7
mdurl==0.1.2
-e git+https://github.com/ROCm/Megatron-LM.git@f40e2768be6146b6104b9f3c26bb08d3054a6c86#egg=megatron_core
mistune==3.1.4
ml_dtypes==0.5.3
mpmath==1.3.0
msgpack==1.1.1
multidict==6.6.4
multiprocess==0.70.16
mypy==1.16.0
mypy_extensions==1.1.0
nbclient==0.10.2
nbconvert==7.16.6
nbformat==5.10.4
nest-asyncio==1.6.0
networkx==2.8.8
ninja==1.11.1.3
nltk==3.9.1
notebook_shim==0.2.4
numba==0.61.2
numcodecs==0.13.1
numpy==1.26.4
nvidia-ml-py==13.580.82
nvidia-modelopt==0.35.0
omegaconf==2.3.0
onnx==1.18.0
onnx-ir==0.1.6
onnxscript==0.3.1
opt-einsum==3.3.0
optree==0.17.0
orjson==3.11.3
overrides==7.7.0
packaging==25.0
pandas==2.2.3
pandocfilters==1.5.1
parameterized==0.8.1
parso==0.8.5
partd==1.4.2
pathspec==0.12.1
peft==0.17.1
pexpect==4.9.0
pfzy==0.3.4
pillow==11.0.0
platformdirs==4.3.8
pluggy==1.6.0
ply==3.11
pooch==1.8.2
prometheus_client==0.23.1
prompt_toolkit==3.0.52
propcache==0.3.2
protobuf==5.29.4
psutil==7.0.0
ptyprocess==0.7.0
PuLP==2.9.0
pure_eval==0.2.3
pwlf==2.2.1
pyarrow==21.0.0
pybind11==3.0.1
pycparser==2.22
pycryptodomex==3.23.0
pydantic==2.11.7
pydantic-settings==2.11.0
pydantic_core==2.33.2
pyDOE==0.3.8
PyGithub==2.3.0
Pygments==2.15.0
PyJWT==2.10.1
PyNaCl==1.5.0
pynvml==13.0.1
pyogrio==0.12.1
pyparsing==3.2.3
pyproj==3.7.1
pyproject_hooks==1.2.0
pyre-extensions==0.0.32
pytest==7.3.2
pytest-cov==7.0.0
pytest-cpp==2.3.0
pytest-csv==3.0.0
pytest-flakefinder==1.1.0
pytest-mock==3.15.0
pytest-random-order==1.2.0
pytest-rerunfailures==14.0
pytest-shard==0.1.2
pytest-subtests==0.13.1
pytest-xdist==3.3.1
python-dateutil==2.9.0.post0
python-dotenv==1.1.1
python-json-logger==3.3.0
pytz==2025.2
PyWavelets==1.4.1
PyYAML==6.0.2
pyzmq==27.1.0
pyzstd==0.17.0
redis==6.4.0
referencing==0.36.2
regex==2025.9.1
requests==2.32.5
requests-toolbelt==1.0.0
rfc3339-validator==0.1.4
rfc3986-validator==0.1.1
rfc3987-syntax==1.1.0
rich==14.1.0
rpds-py==0.27.1
s3transfer==0.10.4
safetensors==0.6.2
scikit-build==0.18.1
scikit-image==0.22.0
scikit-learn==1.7.1
scipy==1.14.1
seaborn==0.13.2
Send2Trash==1.8.3
sentencepiece==0.2.1
sentry-sdk==2.37.1
setuptools-git-versioning==2.1.0
shapely==2.1.2
shtab==1.7.2
simplejson==3.20.1
six==1.17.0
smmap==5.0.2
sniffio==1.3.1
sortedcontainers==2.4.0
soundfile==0.13.1
soupsieve==2.8
soxr==0.5.0.post1
SQLAlchemy==2.0.43
stack-data==0.6.3
sympy==1.14.0
tabulate==0.9.0
tblib==3.1.0
tenacity==9.1.2
tensorboard==2.18.0
tensorboard-data-server==0.7.2
tensorstore==0.1.45
terminado==0.18.1
threadpoolctl==3.6.0
thriftpy2==0.5.3
tifffile==2025.5.10
tiktoken==0.11.0
timm==1.0.26
tinycss2==1.4.0
tlparse==0.3.30
tokenizers==0.21.4
tomli==2.2.1
toolz==1.0.0
torch @ file:///torch-2.9.0.dev20250821%2Brocm7.0.0.lw.git125803b7-cp310-cp310-linux_x86_64.whl#sha256=acab3b73cbed252fa9a561efef45ae39d95bd96d435b2eed6b9c6469b31cd85a
torchao @ file:///workspace/ao
torchaudio @ file:///torchaudio-2.8.0%2Brocm7.0.0.git0c223473-cp310-cp310-linux_x86_64.whl#sha256=96a78ebcc8062af30ccdb00aa4b3d1bbf087a77958e3c22a3efaeeace5049492
torchdata==0.11.0
torchprofile==0.0.4
-e git+https://github.com/AMD-AIG-AIMA/torchtitan.git@04227ff1d8e6f78a2db3a91ef4f55d4686733a2e#egg=torchtitan
-e git+https://github.com/pytorch/torchtune.git@b4c98ac2a37f0397d64c22579aed415ce7264db6#egg=torchtune
torchvision @ file:///torchvision-0.24.0%2Brocm7.0.0.git98f8b375-cp310-cp310-linux_x86_64.whl#sha256=4632280c7fc10b253dae78e9a88af74378fedc044c86ff883f559bb69d15a1c6
tornado==6.5.2
tqdm==4.67.1
traitlets==5.14.3
transformer_engine @ file:///workspace/TransformerEngine
transformers==4.55.0
triton @ file:///triton-3.4.0%2Brocm7.0.0.git56765e8c-cp310-cp310-linux_x86_64.whl#sha256=a4c30fb47fc60d89b8ca0cdc4519f7ebf0f688a03694e10325b285de337191f0
trl==0.21.0
typeguard==4.4.4
types-python-dateutil==2.9.0.20250822
typing-inspect==0.9.0
typing-inspection==0.4.1
typing_extensions==4.14.1
tyro==0.9.31
tzdata==2025.2
ujson==5.11.0
unittest-xml-reporting==3.2.0
uri-template==1.3.0
urllib3==2.5.0
uv==0.8.12
wandb==0.21.4
wcwidth==0.2.13
webcolors==24.11.1
webencodings==0.5.1
websocket-client==1.8.0
Werkzeug==3.1.3
widgetsnbextension==4.0.14
wrapt==1.17.3
xdoctest==1.1.0
xxhash==3.5.0
yappi==1.6.10
yarl==1.20.1
z3-solver==4.15.1.0
zarr==2.18.3
zict==3.0.0
zipp==3.23.0
zstandard==0.25.0
GPU: AMD MI300X
VRAM: 191.7 GB
ROCm: 7.0