diff --git a/torchx/distributed/__init__.py b/torchx/distributed/__init__.py
index 35324e137..8e34f383e 100644
--- a/torchx/distributed/__init__.py
+++ b/torchx/distributed/__init__.py
@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 """
 Convenience methods to use ``torch.distributed``.
 """
diff --git a/torchx/distributed/test/dist_test.py b/torchx/distributed/test/dist_test.py
index 0d53515b4..ccadce695 100644
--- a/torchx/distributed/test/dist_test.py
+++ b/torchx/distributed/test/dist_test.py
@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 import os
 import unittest
 from pathlib import Path
diff --git a/torchx/examples/apps/compute_world_size/config/__init__.py b/torchx/examples/apps/compute_world_size/config/__init__.py
index 2e41cd717..581f84e46 100644
--- a/torchx/examples/apps/compute_world_size/config/__init__.py
+++ b/torchx/examples/apps/compute_world_size/config/__init__.py
@@ -3,3 +3,5 @@
 #
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
+
+# pyre-strict