API

Layers

linear_layer.LinearLayer(in_features, ...[, ...])	Linear layer with support for output weight subsetting.
stacked_linear_layer.StackedLinearLayer(...)	A parallel stacked linear layer that applies multiple linear transformations in parallel.