diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 17e468b454..948693c3c4 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -313,11 +313,11 @@ jobs:
 
       - name: Build package
         if: matrix.os != 'ubuntu-20-16-cores'
-        run: cargo build --release --bin ${{ matrix.file_name }}
+        run: cargo build --release --bin ${{ matrix.file_name }} --features "onnx"
 
-      - name: Build package for ubuntu (with kafka & snowflake)
+      - name: Build package for ubuntu (with kafka & snowflake & onnx)
         if: matrix.os == 'ubuntu-20-16-cores'
-        run: cargo build --release --bin  ${{ matrix.file_name }} --features "kafka snowflake"
+        run: cargo build --release --bin  ${{ matrix.file_name }} --features "kafka snowflake onnx"
 
       - name: Prepare release assets
         shell: bash
diff --git a/ci/Dockerfile b/ci/Dockerfile
index dbd61d5ac8..9237919e5f 100644
--- a/ci/Dockerfile
+++ b/ci/Dockerfile
@@ -6,7 +6,9 @@ RUN apt-get update \
       odbcinst \
       unixodbc \
       curl \
-      unzip
+      unzip \
+      cmake \
+      clang
 
 # INSTALL PROTOBUF
 RUN curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v3.18.1/protoc-3.18.1-linux-x86_64.zip 
@@ -18,6 +20,13 @@ RUN curl -LO https://sfc-repo.snowflakecomputing.com/odbc/linux/2.25.7/snowflake
 RUN dpkg -i snowflake-odbc-2.25.7.x86_64.deb
 RUN rm snowflake-odbc-2.25.7.x86_64.deb
 
+# Install ONNX Runtime
+RUN curl -LO https://github.com/microsoft/onnxruntime/releases/download/v1.15.1/onnxruntime-linux-x64-1.15.1.tgz
+RUN tar -xf onnxruntime-linux-x64-1.15.1.tgz -C /usr/local
+RUN rm onnxruntime-linux-x64-1.15.1.tgz
+RUN cp /usr/local/onnxruntime-linux-x64-1.15.1/lib/libonnxruntime.so.1.15.1 /usr/local/lib/
+RUN ldconfig
+
 # INSTALL DOZER 
 RUN echo "Installing dozer binary"