diff --git a/torchrec/inference/include/torchrec/inference/GPUExecutor.h b/torchrec/inference/include/torchrec/inference/GPUExecutor.h
index b55dacc5a..00c93668b 100644
--- a/torchrec/inference/include/torchrec/inference/GPUExecutor.h
+++ b/torchrec/inference/include/torchrec/inference/GPUExecutor.h
@@ -58,7 +58,7 @@ class GPUExecutor {
       std::shared_ptr<IGPUExecutorObserver>
           observer, // shared_ptr because used in completion executor callback
       std::function<void()> warmupFn = {},
-      std::optional<size_t> numThreadsPerGPU = c10::nullopt,
+      std::optional<size_t> numThreadsPerGPU = std::nullopt,
       std::unique_ptr<GCConfig> gcConfig = std::make_unique<GCConfig>());
   GPUExecutor(GPUExecutor&& executor) noexcept = default;
   GPUExecutor& operator=(GPUExecutor&& executor) noexcept = default;
diff --git a/torchrec/inference/include/torchrec/inference/Validation.h b/torchrec/inference/include/torchrec/inference/Validation.h
index 4844c9ca1..74a2f20ff 100644
--- a/torchrec/inference/include/torchrec/inference/Validation.h
+++ b/torchrec/inference/include/torchrec/inference/Validation.h
@@ -20,7 +20,7 @@ namespace torchrec {
 bool validateSparseFeatures(
     at::Tensor& values,
     at::Tensor& lengths,
-    std::optional<at::Tensor> maybeWeights = c10::nullopt);
+    std::optional<at::Tensor> maybeWeights = std::nullopt);
 
 // Returns whether dense features are valid.
 // Currently validates:
diff --git a/torchrec/inference/inference_legacy/include/torchrec/inference/GPUExecutor.h b/torchrec/inference/inference_legacy/include/torchrec/inference/GPUExecutor.h
index b55dacc5a..00c93668b 100644
--- a/torchrec/inference/inference_legacy/include/torchrec/inference/GPUExecutor.h
+++ b/torchrec/inference/inference_legacy/include/torchrec/inference/GPUExecutor.h
@@ -58,7 +58,7 @@ class GPUExecutor {
       std::shared_ptr<IGPUExecutorObserver>
           observer, // shared_ptr because used in completion executor callback
       std::function<void()> warmupFn = {},
-      std::optional<size_t> numThreadsPerGPU = c10::nullopt,
+      std::optional<size_t> numThreadsPerGPU = std::nullopt,
       std::unique_ptr<GCConfig> gcConfig = std::make_unique<GCConfig>());
   GPUExecutor(GPUExecutor&& executor) noexcept = default;
   GPUExecutor& operator=(GPUExecutor&& executor) noexcept = default;
diff --git a/torchrec/inference/inference_legacy/include/torchrec/inference/Validation.h b/torchrec/inference/inference_legacy/include/torchrec/inference/Validation.h
index 4844c9ca1..74a2f20ff 100644
--- a/torchrec/inference/inference_legacy/include/torchrec/inference/Validation.h
+++ b/torchrec/inference/inference_legacy/include/torchrec/inference/Validation.h
@@ -20,7 +20,7 @@ namespace torchrec {
 bool validateSparseFeatures(
     at::Tensor& values,
     at::Tensor& lengths,
-    std::optional<at::Tensor> maybeWeights = c10::nullopt);
+    std::optional<at::Tensor> maybeWeights = std::nullopt);
 
 // Returns whether dense features are valid.
 // Currently validates: