while_loop

Auto-generated from flax_nnx_code_defs

Keras

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

TensorFlow

API: tf.while_loop
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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