Bump version from 0.2.0 to 0.2.1

Author: ASKabalan

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jax_grid_search"
-version = "0.2.0"
+version = "0.2.1"
 description = "Distributed grid search in JAX"
 authors = [
     { name="Wassim Kabalan" }