fori_loop ========= Auto-generated from flax_nnx_code_defs .. raw:: html
keras.ops.fori_loop
nnx.transforms.iteration.fori_loop