core.models.retro

Exports:

• RetroConfig: configuration dataclass for RetroModel.

• RetroModel: The Retro model.

• get_retro_decoder_block_spec: Get spec for Retro decoder transformer block.

Submodules

• core.models.retro.encoder_attention
• core.models.retro.model
• core.models.retro.decoder_spec
• core.models.retro.encoder_spec
• core.models.retro.base_attention
• core.models.retro.config
• core.models.retro.decoder_attention
• core.models.retro.utils