CombineMasks ============ Combine attention masks using logical AND. **Abstract Signature:** ``CombineMasks(masks: Array, dtype: Dtype = float32)`` .. raw:: html

PyTorch

API:
Strategy: Plugin (combine_masks_logic)

JAX (Core)

API: flax.linen.combine_masks
Strategy: Direct Mapping

Keras

API:
Strategy: Plugin (combine_masks_logic)

Apple MLX

API:
Strategy: Plugin (combine_masks_logic)

Flax NNX

API: flax.nnx.combine_masks
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.combine_masks
Strategy: Direct Mapping