Skip to content

Pull requests: google/hackable_diffusion

New pull request New
14 Open 54 Closed
14 Open 54 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 benchmarks and verification scripts for Hackable Diffusion optimizations.
#68 opened May 1, 2026 by copybara-service Bot Loading…
Refactor Hackable Diffusion for peak XLA performance and Flash Attention support.
#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.