Scan

Functional loop operation carrying state across iterations.

Abstract Signature:

Scan(f: Callable, init, xs)

JAX (Core)

API: jax.lax.scan
Strategy: Direct Mapping
Official Docs ↗

Keras

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

TensorFlow

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

Flax NNX

API: flax.nnx.scan
Strategy: Direct Mapping
Official Docs ↗