Skip to content

[SDY][MaxDiffusion][cleanup] Remove obsolete jax_use_shardy_partitioner=False updates in attention_flax.py - #522

Draft
mbohlool wants to merge 1 commit into
mainfrom
remove-shardy-opt-outs
Draft

[SDY][MaxDiffusion][cleanup] Remove obsolete jax_use_shardy_partitioner=False updates in attention_flax.py#522
mbohlool wants to merge 1 commit into
mainfrom
remove-shardy-opt-outs

Commits

  1. Commits on Nov 12, 2026