Skip to content

Pull requests: google/hackable_diffusion

New pull request New
Clear current search query, filters, and sorts
16 Open 79 Closed
16 Open 79 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

Adjust test tolerances in manifolds_test.py.
#95 by copybara-service Bot was merged May 6, 2026 Loading…
Increase tolerance in manifold geodesic test
#94 by copybara-service Bot was merged May 6, 2026 Loading…
Cleanup time_scheduling - Drop unused class and split tests.
#93 by copybara-service Bot was merged May 6, 2026 Loading…
Update Documentation.
#92 by copybara-service Bot was merged May 6, 2026 Loading…
Cleanup dead code in gaussian_loss.
#91 by copybara-service Bot was merged May 6, 2026 Loading…
Cleanup *_step_sampler.py - fix some interfaces and use keyword arguments consistently
#90 by copybara-service Bot was merged May 6, 2026 Loading…
Add a pointer to kauldron in documentation.
#89 by copybara-service Bot was closed May 6, 2026 Loading…
Add UpdateConditioningFn protocol to base.py for pluggable
#88 by copybara-service Bot was merged May 6, 2026 Loading…
Fix logSNR docstring errors in gaussian.py
#81 by copybara-service Bot was merged May 6, 2026 Loading…
Unify MARK comment casing to Title Case
#80 by copybara-service Bot was merged May 6, 2026 Loading…
Fix safe_norm gradient and RescaledRemaskingFn default
#79 by copybara-service Bot was closed May 6, 2026 Loading…
Small fixes
#78 by copybara-service Bot was closed May 6, 2026 Loading…
Fix duplicate dict key in dit_blocks_test
#77 by copybara-service Bot was closed May 6, 2026 Loading…
Fix misleading docstrings in DiffusionStep and gaussian_loss + removed unused dependencies in base.py
#76 by copybara-service Bot was closed May 6, 2026 Loading…
Fix MultiModalDiffusionNetwork.__call__ signature
#75 by copybara-service Bot was merged May 6, 2026 Loading…
Fix PRNG key reuse, Heun aux mutation, and empty-tree crash
#74 by copybara-service Bot was closed May 6, 2026 Loading…
Add input validation to UniformStratifiedTimeSampler
#73 by copybara-service Bot was merged May 6, 2026 Loading…
Add input validation to EDMTimeSchedule
#72 by copybara-service Bot was merged May 6, 2026 Loading…
Fix typos and docstring errors in hackable_diffusion
#71 by copybara-service Bot was merged May 4, 2026 Loading…
Make conditioning to be able to be arbitrary
#70 by copybara-service Bot was merged May 4, 2026 Loading…
Add logits_dtype to discrete samplers logits and remove jnp.inf
#69 by copybara-service Bot was merged May 4, 2026 Loading…
Add benchmarks and verification scripts for Hackable Diffusion optimizations.
#68 by copybara-service Bot was closed May 4, 2026 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 by copybara-service Bot was merged May 6, 2026 Loading…
Unify discrete samplers with 3-way routing representation.
#63 by copybara-service Bot was merged May 6, 2026 Loading…
Move time_sampling.py into a training module.
#62 by copybara-service Bot was merged Apr 30, 2026 Loading…
ProTip! What’s not been updated in a month: updated:<2026-04-06.