github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-02-12 23:45:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix compatibility list (#4731)

Browse Source
This commit is contained in:
Istvan Kiss
2025-05-13 16:26:36 +02:00
committed by GitHub
parent ea1072b11d
commit f65e1412df
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/compatibility/ml-compatibility/jax-compatibility.rst
View File

@@ -506,12 +506,14 @@ the listed supported JAX versions.
* - Feature
- Description
* - Mixed Precision with TF32
- Mixed precision with TF32 is used for matrix multiplications,
convolutions, and other linear algebra operations, particularly in
deep learning workloads like CNNs and transformers.
* - XLA int4 support
- 4-bit integer (int4) precision in the XLA compiler.
* - MOSAIC (GPU)
- Mosaic is a library of kernel-building abstractions for JAX's Pallas system
- Not Supported