[JAX-Q] Distributed MXFP8 flax layer tests#1643

Merged

phu0ngng merged 2 commits intoNVIDIA:mainfrom

jberchtold-nvidia:dev/jberchtold/mxfp8-layer-test

Apr 4, 2025

Commits on Apr 3, 2025

MXFP8 flax layer tests
jberchtold-nvidia
committed

Commits on Apr 4, 2025

Merge branch 'main' into dev/jberchtold/mxfp8-layer-test
phu0ngng
authored