Skip to content

[JAX] Optimize MoE block - #3354

Open
jberchtold-nvidia wants to merge 48 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/moeblock-debug
Open

[JAX] Optimize MoE block#3354
jberchtold-nvidia wants to merge 48 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/moeblock-debug

Commits

Commits on May 26, 2026

Commits on May 28, 2026

Commits on Jun 1, 2026

Commits on Jun 2, 2026

Commits on Jul 16, 2026

Commits on Jul 17, 2026

Commits on Jul 20, 2026

Commits on Jul 21, 2026

Commits on Jul 27, 2026

Commits on Jul 29, 2026

Commits on Jul 30, 2026

Commits on Jul 31, 2026

Commits on Aug 10, 2026

Commits on Aug 12, 2026

Commits on Aug 13, 2026

Commits on Aug 14, 2026