mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-03 19:05:35 -05:00
* Add JAX compatibility (cherry picked from commit 99215ab6b4cf6a1209d6c5fc781b5855251dcba5) * WIP (cherry picked from commit 54564a85d340b4149ed80a33377cf54c1eb48713) * Fix docker table (cherry picked from commit 8115a905764c869b390de2561e5f1356ec7e9743) * WIP (cherry picked from commit 45076e1fd20fd2c43f7a0ab6d8d5d246c498d801) * add minor formatting (cherry picked from commit c75706841092006c26766611b0407b79a13c7345) * PR feedbacks (cherry picked from commit 236b5daae4251c26cd697c6e20d5982771b05754) * fix inconsistent formatting (cherry picked from commit 0c6a2e3627f9e6159e3f400ab18769904c18097e) * Rename file (cherry picked from commit f17239aa8a9fa1ecdf8dab08c0348dc9216c5311) * jax_triton supported (cherry picked from commit fa56d697fbaa44c0c480df71dc236be8584291c0) * WIP (cherry picked from commit e8f0c5741fe96bb1e3272365906334d911a9a849) * WIP (cherry picked from commit 8ee4f3c62da8e11eea591340dc7c9fc1be8b7035) * WIP (cherry picked from commit 58c6bf441054fe3a21ba2d86808279e90de847b7) * WIP (cherry picked from commit 368ddf6925215a9bfd75a43c7c33def12238f81d) * update .wordlist.txt (cherry picked from commit 78ac332c8d6eba93e2b3e57440da3f60054bbadb) * update .wordlist.txt (cherry picked from commit 8d9492399f4b73b0c3c5359684d5b7faa328ba0f) * Fix typos (cherry picked from commit 394dede13b6de087237832fe3c693c11da7d733b) * update jax note (cherry picked from commit ceacc713c4295f8bbd20fc622579de9053b73337) * Update docs/compatibility/ml-compatibility/jax-compatibility.rst (cherry picked from commit b0613e914a2ba639fddea62eb495f97beaa8ba49) * Update docs/compatibility/ml-compatibility/jax-compatibility.rst (cherry picked from commit 8aac4344b6fd4120a3b8a31878f5316df99f3f99) * Add back hipGraph support (cherry picked from commit 028ddb3535073e0cd668c24614a0a73a491b5948) * WIP (cherry picked from commit 2e0ff9c5e3f88ceea6b0ca770bb4edb52ce08a47) * WIP (cherry picked from commit 186802585de5b7d58f9ac2a7947a83c037df1617) * add blurb about docker icon (cherry picked from commit aef650d4072578f75e7549151613f390f6545ce1) * update pytorch-compatibility path in conf.py * words --------- Co-authored-by: Mátyás Aradi <matyas@streamhpc.com> Co-authored-by: Istvan Kiss <neon60@gmail.com>