Just read the original paper "Attention Is All You Need". Cohere's founder was the guy who wrote that paper in the first place.