expose memory efficient attention functions in __init__.py

Author: lucidrains

memory_efficient_attention_pytorch/__init__.py

@@ -1,2 +1,2 @@
-from memory_efficient_attention_pytorch.memory_efficient_attention import Attention
-from memory_efficient_attention_pytorch.memory_efficient_cosine_sim_attention import CosineSimAttention
+from memory_efficient_attention_pytorch.memory_efficient_attention import Attention, memory_efficient_attention
+from memory_efficient_attention_pytorch.memory_efficient_cosine_sim_attention import CosineSimAttention, numerically_unstable_memory_efficient_attention

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'memory-efficient-attention-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.10',
+  version = '0.0.11',
   license='MIT',
   description = 'Memory Efficient Attention - Pytorch',
   author = 'Phil Wang',