diff --git a/mostlyai/engine/__init__.py b/mostlyai/engine/__init__.py
index a087db6e..ed03937e 100644
--- a/mostlyai/engine/__init__.py
+++ b/mostlyai/engine/__init__.py
@@ -34,7 +34,7 @@
     "TabularARGN",
     "LanguageModel",
 ]
-__version__ = "2.3.3"
+__version__ = "2.4.0"
 
 # suppress specific warning related to os.fork() in multi-threaded processes
 warnings.filterwarnings("ignore", category=DeprecationWarning, message=".*multi-threaded.*fork.*")
diff --git a/pyproject.toml b/pyproject.toml
index ca5b824b..06ce21c5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "mostlyai-engine"
-version = "2.3.3"
+version = "2.4.0"
 description = "Synthetic Data Engine"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"