diff --git a/README.md b/README.md
index 9c855a6..96be821 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # Attention Gym
 
-Attention Gym is a collection of helpful tools and examples for working with flex-attention
+Attention Gym is a collection of helpful tools and examples for working with [flex-attention](https://pytorch.org/docs/main/nn.attention.flex_attention.html#module-torch.nn.attention.flex_attention)
 
 [**🎯 Features**](#-features) |
 [**🚀 Getting Started**](#-getting-started) |