flax
Handle pure bodies in nnx.fori_loop
#5141
Open

Handle pure bodies in nnx.fori_loop #5141

samanklesaria wants to merge 1 commit into google:main from samanklesaria:issues/5113
samanklesaria
samanklesaria Handle pure bodies in nnx.fori_loop
7255c612
samanklesaria samanklesaria requested a review from cgarciae cgarciae 16 days ago
samanklesaria
samanklesaria commented on 2025-12-17
vfdev-5
samanklesaria

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone