diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml
index 0174ac6..780c779 100644
--- a/.github/workflows/build-and-test.yml
+++ b/.github/workflows/build-and-test.yml
@@ -139,7 +139,6 @@ jobs:
         cd third_party/iree
         git submodule update --init -- third_party/googletest
         git submodule update --init -- third_party/flatcc
-        git submodule update --init -- third_party/libyaml
 
     - name: Cache Toolchain
       id: cache-toolchain
diff --git a/build_tools/update_iree_submodules.sh b/build_tools/update_iree_submodules.sh
index 1537fc3..be5972c 100755
--- a/build_tools/update_iree_submodules.sh
+++ b/build_tools/update_iree_submodules.sh
@@ -21,7 +21,6 @@ fi
 cd ${PATH_TO_IREE}
 git submodule update --init -- third_party/googletest
 git submodule update --init -- third_party/flatcc
-git submodule update --init -- third_party/libyaml
 cd ${ORIGINAL_PATH}
 
 echo "Updated the required IREE submodules."
diff --git a/iree-release-link.txt b/iree-release-link.txt
index e1630d7..6c20aae 100644
--- a/iree-release-link.txt
+++ b/iree-release-link.txt
@@ -1 +1 @@
-https://github.com/openxla/iree/releases/download/candidate-20231103.696/iree-dist-20231103.696-linux-x86_64.tar.xz
+https://github.com/openxla/iree/releases/download/candidate-20240222.809/iree-dist-20240222.809-linux-x86_64.tar.xz
diff --git a/requirements-compiler.txt b/requirements-compiler.txt
index b93063a..a4c3d43 100644
--- a/requirements-compiler.txt
+++ b/requirements-compiler.txt
@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2023 Fraunhofer-Gesellschaft zur Förderung der angewandten Forschung e.V.
 # SPDX-License-Identifier: CC0-1.0
 -f https://openxla.github.io/iree/pip-release-links.html
-iree-compiler==20231103.696
+iree-compiler==20240222.809
diff --git a/requirements-tools.txt b/requirements-tools.txt
index 5e6ca64..e6c579a 100644
--- a/requirements-tools.txt
+++ b/requirements-tools.txt
@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2022 Fraunhofer-Gesellschaft zur Förderung der angewandten Forschung e.V.
 # SPDX-License-Identifier: CC0-1.0
 -f https://openxla.github.io/iree/pip-release-links.html
-iree-tools-tf==20231103.696
-iree-tools-tflite==20231103.696
+iree-tools-tf==20240222.809
+iree-tools-tflite==20240222.809
diff --git a/samples/simple_embedding/simple_embedding_float.c b/samples/simple_embedding/simple_embedding_float.c
index 1ea23d3..d3d444c 100644
--- a/samples/simple_embedding/simple_embedding_float.c
+++ b/samples/simple_embedding/simple_embedding_float.c
@@ -40,9 +40,9 @@ iree_status_t Run() {
   IREE_RETURN_IF_ERROR(create_sample_device(iree_allocator_system(), &device),
                        "create device");
   iree_vm_module_t* hal_module = NULL;
-  IREE_RETURN_IF_ERROR(
-      iree_hal_module_create(instance, device, IREE_HAL_MODULE_FLAG_SYNCHRONOUS,
-                             iree_allocator_system(), &hal_module));
+  IREE_RETURN_IF_ERROR(iree_hal_module_create(
+      instance, /*device_count=*/1, &device, IREE_HAL_MODULE_FLAG_SYNCHRONOUS,
+      iree_allocator_system(), &hal_module));
 
   // Load bytecode module from the embedded data.
   const iree_const_byte_span_t module_data = load_bytecode_module_data();
diff --git a/samples/simple_embedding/simple_embedding_int.c b/samples/simple_embedding/simple_embedding_int.c
index 0e63842..b49b118 100644
--- a/samples/simple_embedding/simple_embedding_int.c
+++ b/samples/simple_embedding/simple_embedding_int.c
@@ -40,9 +40,9 @@ iree_status_t Run() {
   IREE_RETURN_IF_ERROR(create_sample_device(iree_allocator_system(), &device),
                        "create device");
   iree_vm_module_t* hal_module = NULL;
-  IREE_RETURN_IF_ERROR(
-      iree_hal_module_create(instance, device, IREE_HAL_MODULE_FLAG_SYNCHRONOUS,
-                             iree_allocator_system(), &hal_module));
+  IREE_RETURN_IF_ERROR(iree_hal_module_create(
+      instance, /*device_count=*/1, &device, IREE_HAL_MODULE_FLAG_SYNCHRONOUS,
+      iree_allocator_system(), &hal_module));
 
   // Load bytecode module from the embedded data.
   const iree_const_byte_span_t module_data = load_bytecode_module_data();
diff --git a/samples/simple_vec_mul/simple_mul_int.c b/samples/simple_vec_mul/simple_mul_int.c
index 72bcc40..e316263 100644
--- a/samples/simple_vec_mul/simple_mul_int.c
+++ b/samples/simple_vec_mul/simple_mul_int.c
@@ -81,9 +81,9 @@ iree_status_t Run() {
 #else
   // Create hal_module
   iree_vm_module_t* hal_module = NULL;
-  IREE_RETURN_IF_ERROR(
-      iree_hal_module_create(instance, device, IREE_HAL_MODULE_FLAG_SYNCHRONOUS,
-                             iree_allocator_system(), &hal_module));
+  IREE_RETURN_IF_ERROR(iree_hal_module_create(
+      instance, /*device_count=*/1, &device, IREE_HAL_MODULE_FLAG_SYNCHRONOUS,
+      iree_allocator_system(), &hal_module));
 
   iree_vm_module_t* modules[] = {hal_module, module};
 #endif
diff --git a/third_party/iree b/third_party/iree
index d354465..5d8907e 160000
--- a/third_party/iree
+++ b/third_party/iree
@@ -1 +1 @@
-Subproject commit d354465a12182e59ed21918420b5b777b7a9b6db
+Subproject commit 5d8907e82fc1eb741a4d4d27f5cae865323fd1d7