CombineMasks

Combine attention masks using logical AND.

Abstract Signature:

CombineMasks(masks: Array, dtype: Dtype = float32)

PyTorchJAX (Core)KerasApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Plugin (combine_masks_logic)

JAX (Core)

API: flax.linen.combine_masks

Strategy: Direct Mapping

Official Docs ↗

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

Official Docs ↗

PaxML / Praxis

API: praxis.layers.combine_masks

Strategy: Direct Mapping

Official Docs ↗