diff --git a/.gitignore b/.gitignore
index c8acc8a..2535776 100644
--- a/.gitignore
+++ b/.gitignore
@@ -31,4 +31,5 @@ var/
 # Dev
 venv
 .tup
-tup.config
\ No newline at end of file
+tup.config
+local
\ No newline at end of file
diff --git a/src/flash_attn_jax/__init__.py b/src/flash_attn_jax/__init__.py
index 2174677..8297595 100644
--- a/src/flash_attn_jax/__init__.py
+++ b/src/flash_attn_jax/__init__.py
@@ -1,2 +1,2 @@
 from .flash import flash_mha
-__version__ = 'v0.2.0'
+__version__ = 'v0.2.1'