diff --git a/src/ep/tensorrt.rs b/src/ep/tensorrt.rs
index fe8bb9a..36908b4 100644
--- a/src/ep/tensorrt.rs
+++ b/src/ep/tensorrt.rs
@@ -50,6 +50,12 @@ impl TensorRT {
 		self
 	}
 
+	#[must_use]
+	pub fn with_bf16(mut self, enable: bool) -> Self {
+		self.options.set("trt_bf16_enable", if enable { "1" } else { "0" });
+		self
+	}
+
 	#[must_use]
 	pub fn with_int8(mut self, enable: bool) -> Self {
 		self.options.set("trt_int8_enable", if enable { "1" } else { "0" });