megatron.model.utils.get_linear_layer

megatron.model.utils.get_linear_layer(rows: int, columns: int, init_method: Callable, perform_initialization: bool)

Simple linear layer with weight initialization.