def get_cell_with_dropout(self, rng: RandomStreams, dropout_rate: float): raise NotImplementedError()