pub fn gru( in_dim: usize, hidden_dim: usize, config: GRUConfig, vb: VarBuilder<'_>, ) -> Result<GRU>