Skip to content

Pull requests: google/hackable_diffusion

New pull request New
23 Open 58 Closed
23 Open 58 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

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…
Add input validation to EDMTimeSchedule
#72 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…
Add tests for VelocityStep in gaussian_step_sampler_test.py
#60 opened Apr 29, 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…
Add AdditiveSequenceEmbedding
#12 opened Feb 19, 2026 by copybara-service Bot Loading…
Bump protobuf from 6.33.0 to 6.33.5 in the pip group across 1 directory dependencies Pull requests that update a dependency file python Pull requests that update python code
#8 opened Feb 5, 2026 by dependabot Bot Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.