Skip to content

Extend JAX support to the grouped/discrete-grouped GEMM APIs and proj_rope_mxfp8 (stacked on #529) - #530

Closed
Anerudhan wants to merge 1 commit into
NVIDIA:developfrom
Anerudhan:grouped-unfused-jax
Closed

Extend JAX support to the grouped/discrete-grouped GEMM APIs and proj_rope_mxfp8 (stacked on #529)#530
Anerudhan wants to merge 1 commit into
NVIDIA:developfrom
Anerudhan:grouped-unfused-jax

Extend JAX support to the grouped/discrete-grouped GEMM APIs

6dcef0d
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs