The "attention" mechanism that they propose in their paper does just that. They named the neural network architecture obtained after adding attention a "Transformer". Transformers have transformed AI/ML. ChatGPT is essentially a transformer with some (albeit important) tweaks.