src/torch_dimensions/models/__init__.py

from torch_dimensions.models.rnn import GRU, LSTM

__all__ = ["GRU", "LSTM"]