map

Auto-generated from flax_nnx_code_defs

Keras

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

TensorFlow

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

Flax NNX

API: nnx.statelib.State.map
Strategy: Direct Mapping
Official Docs ↗