WIP

Fix compatibility list (#4731 )
2026-01-11 07:38:17 -05:00 · 2025-05-13 17:03:23 +02:00 · 2025-05-13 16:26:36 +02:00
1 changed files with 4 additions and 2 deletions
--- a/docs/compatibility/ml-compatibility/jax-compatibility.rst
+++ b/docs/compatibility/ml-compatibility/jax-compatibility.rst
@@ -496,7 +496,7 @@ Modules for JAX extensions.
      - 5.5.0

 Unsupported JAX features
--------------------------------------------------------------------------------
+===============================================================================

 The following GPU-accelerated JAX features are not supported by ROCm for
 the listed supported JAX versions.
@@ -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
Author	SHA1	Message	Date
Istvan Kiss	e72d89c589	WIP	2025-05-13 17:03:23 +02:00
Istvan Kiss	f65e1412df	Fix compatibility list (#4731 )	2025-05-13 16:26:36 +02:00