From f65e1412df66d1d96e54f2f45dacf56c9aed57de Mon Sep 17 00:00:00 2001 From: Istvan Kiss Date: Tue, 13 May 2025 16:26:36 +0200 Subject: [PATCH] Fix compatibility list (#4731) --- docs/compatibility/ml-compatibility/jax-compatibility.rst | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docs/compatibility/ml-compatibility/jax-compatibility.rst b/docs/compatibility/ml-compatibility/jax-compatibility.rst index 471b2efc3..ba4b3aad4 100644 --- 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