StochasticResidual

Stochastic residual layer that randomly drops the residual branch.

Abstract Signature:

StochasticResidual(residual_weight: float = 1.0, survival_prob: float = 1.0)

PyTorch

API: torchvision.ops.StochasticDepth
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.linen.StochasticDepth
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: praxis.layers.StochasticResidual
Strategy: Direct Mapping
Official Docs ↗