diff --git a/torch/extension.h b/torch/extension.h
index 828aefd..0ff1425 100644
--- a/torch/extension.h
+++ b/torch/extension.h
@@ -1,6 +1,8 @@
 #pragma once
 
-#include <Python.h>
-
-#include <torch/csrc/utils/pybind.h>
+// All pure C++ headers for the C++ frontend.
+#include <torch/all.h>
+// Python bindings for the C++ frontend (includes Python.h).
+#include <torch/python.h>
+// Deprecated tensor factories (to be removed).
 #include <torch/csrc/variable_tensor_functions.h>