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