mlx.nn.TransformerDecoder

class TransformerDecoder(num_layers: int, dims: int, num_heads: int, mlp_dims: int | None = None, dropout: float = 0.0, activation=<mlx.gc_func object>, norm_first: bool = True, checkpoint: bool = False)

Methods