-record(encoder_block, {
    attention :: viva_tensor@nn@attention:multi_head_attention(),
    ffn :: viva_tensor@nn@transformer:feed_forward(),
    norm1 :: viva_tensor@nn@norm:layer_norm(),
    norm2 :: viva_tensor@nn@norm:layer_norm()
}).