diff --git a/poetry.lock b/poetry.lock
index 9f99e15ee3..233ac74514 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9d7c148fcbb968b9a1fc8a5bbba6baaf14a8e80decc5da3c9dc3d88c6758c8b7
-size 778425
+oid sha256:740934eaa6ba344c6589cdcf68f0e791fac2298992355d4005410e43fc31744f
+size 778659
diff --git a/pyproject.toml b/pyproject.toml
index 542f138921..de77b4ea41 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -175,6 +175,7 @@ tritonclient = {version = "2.28.0", extras = ["http"]}
 transformers = "*"
 timm = "==0.9.2"
 PyNvCodec = { git = "https://github.com/NVIDIA/VideoProcessingFramework.git", rev = "3347e55" }
+apex = { url = "https://github.com/commaai/apex/releases/download/pytorch2.0.1%2Bcu11.8/apex-0.1-cp311-cp311-linux_x86_64.whl" }
 
 
 [build-system]