.. Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. See LICENSE for license information. JAX: Expert Parallelism with TransformerEngine ============================================== **TODO — Coming soon.** `← Back to the JAX integration overview <../te_jax_integration.html>`_