TFAttentionBlocks Module

Provides helper functions and classes for building and utilizing attention blocks in TensorFlow models.