PyTorch
API: torch.nn.AlphaDropout
Strategy: Direct Mapping
JAX (Core)
API: flax.linen.AlphaDropout
Strategy: Direct Mapping
Keras
API: keras.layers.AlphaDropout
Strategy: Direct Mapping
TensorFlow
API: keras.layers.AlphaDropout
Strategy: Direct Mapping
Apple MLX
API: mx.nn.AlphaDropout
Strategy: Direct Mapping
Flax NNX
API: flax.linen.AlphaDropout
Strategy: Direct Mapping
PaxML / Praxis
API: praxis.layers.AlphaDropout
Strategy: Macro 'praxis.layers.AlphaDropout(keep_prob=1.0-{p})'