text: """Modules."""
  
  from .attention import MultiHeadAttention
  from .batch_norm import BatchNorm1D, BatchNorm2D
  from .conv import Conv1D, Conv1DTranspose, Conv2D, Conv2DTranspose
  from .dropout import Dropout
  from .ema import EMA
  from .embed import Embed
  from .group_norm import GroupNorm
  from .identity import Identity
  from .lambda_module import Lambda
  from .layer_norm import LayerNorm
  from .linear import Linear
  from .pool import avg_pool, max_pool
  from .recurrent import GRU, LSTM, GRUState, LSTMState, VanillaRNN, VanillaRNNState
  from .rng_seq import RngSeq
  from .sequential import Sequential
  