diff --git a/setup.py b/setup.py
index 867afc1b905..6e9927f35a8 100644
--- a/setup.py
+++ b/setup.py
@@ -484,7 +484,10 @@ def _get_jax_install_requirements():
         *_get_jax_install_requirements(),
     ],
     package_data={
-        'torch_xla': ['lib/*.so*',],
+        'torch_xla': [
+            'lib/*.so*',
+            'py.typed',
+        ],
     },
     entry_points={
         'console_scripts': [
diff --git a/torch_xla/py.typed b/torch_xla/py.typed
new file mode 100644
index 00000000000..e69de29bb2d