corpus.blog
43,228 blogs · 5,519,570 posts Claim your blog
Record · corpus.blog/posts/01a0b0c1-a5f2-717e-8f66-78805769d838

Accelerating Dropless MoE Training in JAX with NVIDIA Transformer Engine

NVIDIA Developer Blog · published 14 September 2026

A record is what was published and who pointed at it. The text of the post is not held here: read it at the source, or at its Wayback capture.

Read the post at developer.nvidia.com Wayback capture
0blogs citing
First cited
—
Most recent
—
Rank this month
not ranked
External links
13
Words captured
2,439

Cited by

No blog we hold has cited this post yet.

Links from this post

Link
Host
https://www.nvidia.com/en-us/glossary/mixture-of-experts/
nvidia.com
https://github.com/NVIDIA/TransformerEngine
github.com
https://arxiv.org/abs/2211.15841
arxiv.org
https://www.nvidia.com/en-us/data-center/technologies/blackwell-architecture/
nvidia.com
https://github.com/NVIDIA/nccl-extensions
github.com
https://www.nvidia.com/en-us/data-center/nvlink/
nvidia.com
https://ghcr.io/nvidia/jax:maxtext-2026-09-10
ghcr.io
https://github.com/NVIDIA/JAX-Toolbox/
github.com
http://ghcr.io/nvidia/jax:maxtext-2026-09-10
ghcr.io
https://github.com/AI-Hypercomputer/maxtext/blob/main/docs/reference/core_concepts/moe_configuration.md
github.com
https://openxla.org/xla/flags_guidance
openxla.org
https://docs.nvidia.com/jax-toolbox/performance-profiling/gpu-performance
docs.nvidia.com
https://docs.nvidia.com/deeplearning/transformer-engine/user-guide/index.html
docs.nvidia.com
Elsewhere on developer.nvidia.com (3)