Skip to content

Pull requests: google/hackable_diffusion

New pull request New
27 Open 61 Closed
27 Open 61 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add UpdateConditioningFn protocol to base.py for pluggable
#87 opened May 6, 2026 by copybara-service Bot Loading…
Add stop-token truncation for discrete samplers
#86 opened May 6, 2026 by copybara-service Bot Loading…
DO NOT SUBMIT
#85 opened May 5, 2026 by copybara-service Bot Loading…
1
Add diagram to discrete_step_sampler.py docstring illustrating routing and planning flow.
#84 opened May 5, 2026 by copybara-service Bot Loading…
Add copy conditioning encoder
#83 opened May 5, 2026 by copybara-service Bot Loading…
Make the conditioning embeddings to have arbitrary str and arbitrary output type.
#82 opened May 5, 2026 by copybara-service Bot Loading…
Fix logSNR docstring errors in gaussian.py
#81 opened May 5, 2026 by copybara-service Bot Loading…
Unify MARK comment casing to Title Case
#80 opened May 5, 2026 by copybara-service Bot Loading…
Fix safe_norm gradient and RescaledRemaskingFn default
#79 opened May 4, 2026 by copybara-service Bot Loading…
Small fixes
#78 opened May 4, 2026 by copybara-service Bot Loading…
Fix duplicate dict key in dit_blocks_test
#77 opened May 4, 2026 by copybara-service Bot Loading…
Fix misleading docstrings in DiffusionStep and gaussian_loss + removed unused dependencies in base.py
#76 opened May 4, 2026 by copybara-service Bot Loading…
Fix MultiModalDiffusionNetwork.__call__ signature
#75 opened May 4, 2026 by copybara-service Bot Loading…
Fix PRNG key reuse, Heun aux mutation, and empty-tree crash
#74 opened May 4, 2026 by copybara-service Bot Loading…
Add input validation to UniformStratifiedTimeSampler
#73 opened May 4, 2026 by copybara-service Bot Loading…
Refactor Hackable Diffusion for improved XLA performance with Flash Attention and fused RMS normalization.
#67 opened May 1, 2026 by copybara-service Bot Loading…
Update documentation for routing and planner architecture.
#66 opened May 1, 2026 by copybara-service Bot Loading…
Add GreedyPlanner matching.
#65 opened May 1, 2026 by copybara-service Bot Loading…
Factor out candidate generation and post-processing in discrete samplers to reduce boilerplate. This makes the update methods focus purely on computing the routing probabilities.
#64 opened Apr 30, 2026 by copybara-service Bot Loading…
Unify discrete samplers with 3-way routing representation.
#63 opened Apr 30, 2026 by copybara-service Bot Loading…
Use jnp.zeros rather than jnp.empty in places where values are important.
#51 opened Apr 15, 2026 by copybara-service Bot Loading…
Add cross attention to DiT
#36 opened Mar 24, 2026 by copybara-service Bot Loading…
Add cross attention to DiT blocks
#34 opened Mar 24, 2026 by copybara-service Bot Loading…
fix: two silent math bugs in IntegratedDiscreteDDIMStep and QK-norm attention scale
#29 opened Mar 15, 2026 by Dhakshin2007 Loading…
Add Diffusion transformer
#13 opened Feb 19, 2026 by copybara-service Bot Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.