Change import structure (#125)

Author: drisspg

attn_gym/__init__.py

@@ -1 +1,6 @@
 from attn_gym.utils import visualize_attention_scores
+import attn_gym.mods
+import attn_gym.masks
+import attn_gym.paged_attention
+
+__all__ = ["visualize_attention_scores", "mods", "masks", "paged_attention"]