Skip to content

Use jax.lax.scaled_dot for MXFP dot_general in Qwix instead of jax.nn.scaled_matmul that binds to cuDNN - #369

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_981031906
Open

Use jax.lax.scaled_dot for MXFP dot_general in Qwix instead of jax.nn.scaled_matmul that binds to cuDNN#369
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_981031906

Use `jax.lax.scaled_dot` for MXFP dot_general in Qwix instead of `ja…

e0f4dd0
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar