| 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 |
|
|
