Components
Dropout

Dropout

A dropout layer that wraps a given Component.

Dropout(nn.Linear(128, 256), p=0.2)

Forward

(x: jaxtyping.Float[Tensor, ' *shape']) -> jaxtyping.Float[Tensor, ' *shape']
AttachDropPath