flax
Add diffusion example ported from Jax AI stack
#5403
Merged

Add diffusion example ported from Jax AI stack #5403

samanklesaria
samanklesaria Add diffusion example
e8a62a5e
review-notebook-app
samanklesaria samanklesaria marked this pull request as draft 35 days ago
samanklesaria samanklesaria force pushed from 0b42300f to 8f6c2183 35 days ago
samanklesaria samanklesaria force pushed from 8f6c2183 to de862604 35 days ago
samanklesaria Use flow matching for diffusion example
0f1ef741
samanklesaria samanklesaria force pushed from de862604 to 0f1ef741 35 days ago
samanklesaria samanklesaria marked this pull request as ready for review 35 days ago
IvyZX
IvyZX approved these changes on 2026-04-13
IvyZX IvyZX added pull ready
copybara-service copybara-service merged 9e73be40 into main 31 days ago
vfdev-5
vfdev-5 commented on 2026-04-16

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone