Fix typo in mjx.rst

This commit is contained in:
Alessandro Croci
2024-12-06 09:16:37 +01:00
committed by GitHub
parent fb8e40208c
commit 342a72839a
+1 -1
View File
@@ -416,4 +416,4 @@ The following environment variables should be set:
``XLA_FLAGS=--xla_gpu_triton_gemm_any=true`` ``XLA_FLAGS=--xla_gpu_triton_gemm_any=true``
This enables the Triton-based GEMM (matmul) emitter for any GEMM that it supports. This can yield a 30% speedup on This enables the Triton-based GEMM (matmul) emitter for any GEMM that it supports. This can yield a 30% speedup on
NVIDIA GPUs. If you have multiple GPUs, you may also benefit from enabling flags related to NVIDIA GPUs. If you have multiple GPUs, you may also benefit from enabling flags related to
`communciation between GPUs <https://jax.readthedocs.io/en/latest/gpu_performance_tips.html>`__. `communication between GPUs <https://jax.readthedocs.io/en/latest/gpu_performance_tips.html>`__.