diff --git a/torchvision/__init__.py b/torchvision/__init__.py
index ce8f9784ca6..1f21d34865d 100644
--- a/torchvision/__init__.py
+++ b/torchvision/__init__.py
@@ -1,5 +1,7 @@
 import warnings
 
+from .extension import _HAS_OPS
+
 from torchvision import models
 from torchvision import datasets
 from torchvision import ops
@@ -7,7 +9,6 @@
 from torchvision import utils
 from torchvision import io
 
-from .extension import _HAS_OPS
 import torch
 
 try:
diff --git a/torchvision/models/detection/transform.py b/torchvision/models/detection/transform.py
index ffbe2279b78..b7c7250c5a0 100644
--- a/torchvision/models/detection/transform.py
+++ b/torchvision/models/detection/transform.py
@@ -223,4 +223,4 @@ def resize_boxes(boxes, original_size, new_size):
     xmax = xmax * ratio_width
     ymin = ymin * ratio_height
     ymax = ymax * ratio_height
-    return torch.stack((xmin, ymin, xmax, ymax), dim=1)
+    return torch.stack((xmin, ymin, xmax, ymax), dim=1)
\ No newline at end of file
diff --git a/torchvision/ops/boxes.py b/torchvision/ops/boxes.py
index 4bbc9c1fdad..1b44a034ba3 100644
--- a/torchvision/ops/boxes.py
+++ b/torchvision/ops/boxes.py
@@ -4,6 +4,7 @@
 import torchvision
 
 
+@torch.jit.script
 def nms(boxes, scores, iou_threshold):
     # type: (Tensor, Tensor, float)
     """
@@ -35,6 +36,7 @@ def nms(boxes, scores, iou_threshold):
     return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
 
 
+@torch.jit.script
 def batched_nms(boxes, scores, idxs, iou_threshold):
     # type: (Tensor, Tensor, Tensor, float)
     """