pub fn linear_b( in_dim: usize, out_dim: usize, bias: bool, vb: ShardedVarBuilder, ) -> Result<Linear>