def METHOD_NAME(self):
    return dict(
        rnn=(
            (
                ("layer", self.num_recurrent_layers),
                ("sampler", None),
                ("hidden", self.recurrent_hidden_state_size),
            ),
            torch.float32,
        )
    )