diff --git a/src/deepsparse/utils/data.py b/src/deepsparse/utils/data.py
index d162c3041f..baaf50b551 100644
--- a/src/deepsparse/utils/data.py
+++ b/src/deepsparse/utils/data.py
@@ -28,6 +28,7 @@
     "split_engine_inputs",
     "join_engine_outputs",
     "prep_for_serialization",
+    "numpy_log_softmax",
 ]
 
 from pydantic import BaseModel