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
Apple MLX
API: mx.nn.AlphaDropout
Strategy: Direct Mapping
Flax NNX
API: flax.linen.AlphaDropout
Strategy: Direct Mapping