fori_loop

Auto-generated from flax_nnx_code_defs

Keras

API: keras.ops.fori_loop
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.transforms.iteration.fori_loop
Strategy: Direct Mapping
Official Docs ↗