swyft
swyft copied to clipboard
LazyLinear etc
We should explore if replacing some of the network structures with Lazy initialization could simplify writing head-networks etc. We might get around having to specify self.n_features in that way.