diff --git a/CHANGELOG.md b/CHANGELOG.md
index 3706a45..28b39bd 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -4,6 +4,9 @@ All notable changes to this project will be documented in this file.
  
 The format is based on [this sample changelog](https://gist.github.com/juampynr/4c18214a8eb554084e21d6e288a18a2c).
  
+## [0.2.4] - 2023-03-31
+- Update Tensorflow and MLflow versions
+
 
 ## [0.2.3] - 2023-02-03
 
diff --git a/setup.py b/setup.py
index 94cc311..2f87027 100644
--- a/setup.py
+++ b/setup.py
@@ -21,9 +21,9 @@
         "requests",
         "networkx==2.6.2",
         "torch==1.13.1",
-        "tensorflow==2.11.0",
+        "tensorflow==2.11.1",
         "tqdm==4.61.2",
-        "mlflow==2.1.1",
+        "mlflow==2.2.1",
         "importlib_resources"
 
     ],
diff --git a/src/graphmb/version.py b/src/graphmb/version.py
index e807d94..07d2fd3 100644
--- a/src/graphmb/version.py
+++ b/src/graphmb/version.py
@@ -1,2 +1,2 @@
-__version__ = "0.2.3"
+__version__ = "0.2.4"