-record(t5_model, {
    embedding :: viva_tensor@nn@embedding:embedding(),
    encoder_blocks :: list(viva_tensor@models@t5:t5_block()),
    decoder_blocks :: list(viva_tensor@models@t5:t5_block()),
    encoder_final_norm :: viva_tensor@nn@norm:rms_norm(),
    decoder_final_norm :: viva_tensor@nn@norm:rms_norm(),
    output_proj :: viva_tensor@tensor:tensor()
}).