Fix compatibility list

2026-01-09 22:58:17 -05:00 · 2025-05-13 16:02:55 +02:00
1 changed files with 3 additions and 1 deletions
--- a/docs/compatibility/ml-compatibility/jax-compatibility.rst
+++ b/docs/compatibility/ml-compatibility/jax-compatibility.rst
@@ -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