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

[SDY][MaxDiffusion][cleanup] Remove obsolete jax_use_shardy_partition…

e922644
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Nov 12, 2026 in 15s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

✅ e922644 Author: @mbohlool <bo****l​@gmail.com>

(Only the first commit for a unique contributor is listed.)

View more details on Google CLA