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