pub fn linear_b( in_dim: usize, out_dim: usize, bias: bool, vb: VarBuilderArgs<'_, ShardedSafeTensors<'_>>, ) -> Result<Linear, Error>