Fix compatibility list

2026-01-10 23:28:03 -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