diff --git a/docs/_modules/torch_tensorrt/_compile.html b/docs/_modules/torch_tensorrt/_compile.html
index 914db0a7c9..fb68a55437 100644
--- a/docs/_modules/torch_tensorrt/_compile.html
+++ b/docs/_modules/torch_tensorrt/_compile.html
@@ -518,7 +518,7 @@ <h1>Source code for torch_tensorrt._compile</h1><div class="highlight"><pre>
                     <span class="c1"># profiling_verbosity=trt.ProfilingVerbosity.DETAILED, #For profile</span>
                 <span class="p">)</span>
                 <span class="c1"># For profile</span>
-                <span class="c1"># from fx2trt_oss.fx.tools.trt_profiler_sorted import profile_trt_module</span>
+                <span class="c1"># from torch_tensorrt.fx.tools.trt_profiler_sorted import profile_trt_module</span>
                 <span class="c1"># profile_trt_module(&quot;&quot;, trt_mod, acc_inputs)</span>
                 <span class="n">trt_mod</span> <span class="o">=</span> <span class="n">TRTModule</span><span class="p">(</span><span class="o">*</span><span class="n">r</span><span class="p">)</span>
 
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
index bb63ef5242..52ca551142 100644
--- a/py/torch_tensorrt/_compile.py
+++ b/py/torch_tensorrt/_compile.py
@@ -172,7 +172,7 @@ def get_input(self, inputs):
                     # profiling_verbosity=trt.ProfilingVerbosity.DETAILED, #For profile
                 )
                 # For profile
-                # from fx2trt_oss.fx.tools.trt_profiler_sorted import profile_trt_module
+                # from torch_tensorrt.fx.tools.trt_profiler_sorted import profile_trt_module
                 # profile_trt_module("", trt_mod, acc_inputs)
                 trt_mod = TRTModule(*r)
 
diff --git a/py/torch_tensorrt/fx/converters/acc_ops_converters.py b/py/torch_tensorrt/fx/converters/acc_ops_converters.py
index 62326ba841..ef757bf3d1 100644
--- a/py/torch_tensorrt/fx/converters/acc_ops_converters.py
+++ b/py/torch_tensorrt/fx/converters/acc_ops_converters.py
@@ -4,22 +4,21 @@
 import warnings
 from typing import cast, Dict, Optional, Sequence, Tuple, Union
 
-from ..tracer.acc_tracer import acc_ops
 import numpy as np
 
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
+
+from ..tracer.acc_tracer import acc_ops
 from ..types import *  # noqa: F403
-from ..utils import (
-    get_dynamic_dims,
-    torch_dtype_from_trt,
-    torch_dtype_to_trt,
-)
 from torch.fx.immutable_collections import immutable_list
 from torch.fx.node import Argument, Target
 
+from ..utils import get_dynamic_dims, torch_dtype_from_trt, torch_dtype_to_trt
+
 from .converter_utils import *  # noqa: F403
 
 
diff --git a/py/torch_tensorrt/fx/converters/activation.py b/py/torch_tensorrt/fx/converters/activation.py
index d53d840fdd..a7ab25152c 100644
--- a/py/torch_tensorrt/fx/converters/activation.py
+++ b/py/torch_tensorrt/fx/converters/activation.py
@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import mark_as_int8_layer
diff --git a/py/torch_tensorrt/fx/converters/adaptive_avgpool.py b/py/torch_tensorrt/fx/converters/adaptive_avgpool.py
index e2cb233b0b..8de9987c77 100644
--- a/py/torch_tensorrt/fx/converters/adaptive_avgpool.py
+++ b/py/torch_tensorrt/fx/converters/adaptive_avgpool.py
@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import extend_mod_attr_to_tuple, mark_as_int8_layer
diff --git a/py/torch_tensorrt/fx/converters/add.py b/py/torch_tensorrt/fx/converters/add.py
index 68d0e8e4c7..c60b0313a3 100644
--- a/py/torch_tensorrt/fx/converters/add.py
+++ b/py/torch_tensorrt/fx/converters/add.py
@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer
diff --git a/py/torch_tensorrt/fx/converters/batchnorm.py b/py/torch_tensorrt/fx/converters/batchnorm.py
index ff27a4e8c3..130991df54 100644
--- a/py/torch_tensorrt/fx/converters/batchnorm.py
+++ b/py/torch_tensorrt/fx/converters/batchnorm.py
@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer, to_numpy
diff --git a/py/torch_tensorrt/fx/converters/converter_utils.py b/py/torch_tensorrt/fx/converters/converter_utils.py
index ef13851fef..c59763b5ab 100644
--- a/py/torch_tensorrt/fx/converters/converter_utils.py
+++ b/py/torch_tensorrt/fx/converters/converter_utils.py
@@ -7,6 +7,8 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+from torch.fx.node import Argument, Target
+
 from ..types import (
     Shape,
     TRTDataType,
@@ -18,7 +20,6 @@
     TRTTensor,
 )
 from ..utils import torch_dtype_from_trt
-from torch.fx.node import Argument, Target
 
 
 def get_trt_plugin(
diff --git a/py/torch_tensorrt/fx/converters/convolution.py b/py/torch_tensorrt/fx/converters/convolution.py
index f2de7cedea..94c1ce24f1 100644
--- a/py/torch_tensorrt/fx/converters/convolution.py
+++ b/py/torch_tensorrt/fx/converters/convolution.py
@@ -2,6 +2,7 @@
 import numpy as np
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import (
diff --git a/py/torch_tensorrt/fx/converters/linear.py b/py/torch_tensorrt/fx/converters/linear.py
index 13597b2cf8..e7cca6f76a 100644
--- a/py/torch_tensorrt/fx/converters/linear.py
+++ b/py/torch_tensorrt/fx/converters/linear.py
@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer, to_numpy
diff --git a/py/torch_tensorrt/fx/converters/maxpool.py b/py/torch_tensorrt/fx/converters/maxpool.py
index 4348218c7b..6c64a3b108 100644
--- a/py/torch_tensorrt/fx/converters/maxpool.py
+++ b/py/torch_tensorrt/fx/converters/maxpool.py
@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import extend_mod_attr_to_tuple, mark_as_int8_layer
diff --git a/py/torch_tensorrt/fx/converters/mul.py b/py/torch_tensorrt/fx/converters/mul.py
index ef5b5d53d0..a1d9858ebd 100644
--- a/py/torch_tensorrt/fx/converters/mul.py
+++ b/py/torch_tensorrt/fx/converters/mul.py
@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer
diff --git a/py/torch_tensorrt/fx/converters/quantization.py b/py/torch_tensorrt/fx/converters/quantization.py
index 1e7ca7a14e..8f6b49cda7 100644
--- a/py/torch_tensorrt/fx/converters/quantization.py
+++ b/py/torch_tensorrt/fx/converters/quantization.py
@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, get_inputs_from_args_and_kwargs
diff --git a/py/torch_tensorrt/fx/converters/transformation.py b/py/torch_tensorrt/fx/converters/transformation.py
index 7ad266a2b1..62cfef8453 100644
--- a/py/torch_tensorrt/fx/converters/transformation.py
+++ b/py/torch_tensorrt/fx/converters/transformation.py
@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import mark_as_int8_layer
diff --git a/py/torch_tensorrt/fx/example/fx2trt_example.py b/py/torch_tensorrt/fx/example/fx2trt_example.py
index 6b331da7be..8c648ec065 100644
--- a/py/torch_tensorrt/fx/example/fx2trt_example.py
+++ b/py/torch_tensorrt/fx/example/fx2trt_example.py
@@ -1,11 +1,11 @@
 # type: ignore[]
 
-import fx2trt_oss.tracer.acc_tracer.acc_tracer as acc_tracer
 import torch
 import torch.fx
 import torch.nn as nn
-from fx2trt_oss.fx import InputTensorSpec, TRTInterpreter, TRTModule
-from fx2trt_oss.fx.tools.trt_splitter import TRTSplitter
+import torch_tensorrt.fx.tracer.acc_tracer.acc_tracer as acc_tracer
+from torch_tensorrt.fx import InputTensorSpec, TRTInterpreter, TRTModule
+from torch_tensorrt.fx.tools.trt_splitter import TRTSplitter
 
 
 # The purpose of this example is to demonstrate the overall flow of lowering a PyTorch
@@ -83,12 +83,12 @@ def forward(self, x):
     %x : [#users=1] = placeholder[target=x]
     %linear_weight : [#users=1] = get_attr[target=linear.weight]
     %linear_bias : [#users=1] = get_attr[target=linear.bias]
-    %linear_1 : [#users=1] = call_function[target=fx2trt_oss.tracer.acc_tracer.acc_ops.linear](args = (), ...
-    %relu_1 : [#users=1] = call_function[target=fx2trt_oss.tracer.acc_tracer.acc_ops.relu](args = (), ...
+    %linear_1 : [#users=1] = call_function[target=torch_tensorrt.fx.tracer.acc_tracer.acc_ops.linear](args = (), ...
+    %relu_1 : [#users=1] = call_function[target=torch_tensorrt.fx.tracer.acc_tracer.acc_ops.relu](args = (), ...
     return relu_1
 graph():
     %relu_1 : [#users=1] = placeholder[target=relu_1]
-    %linalg_norm_1 : [#users=1] = call_function[target=fx2trt_oss.tracer.acc_tracer.acc_ops.linalg_norm](args = (), ...
+    %linalg_norm_1 : [#users=1] = call_function[target=torch_tensorrt.fx.tracer.acc_tracer.acc_ops.linalg_norm](args = (), ...
     return linalg_norm_1
 """
 
diff --git a/py/torch_tensorrt/fx/example/lower_example.py b/py/torch_tensorrt/fx/example/lower_example.py
index 17b2dd4755..b93e93598e 100644
--- a/py/torch_tensorrt/fx/example/lower_example.py
+++ b/py/torch_tensorrt/fx/example/lower_example.py
@@ -4,8 +4,8 @@
 
 import torch
 import torchvision
-from fx2trt_oss.fx.lower import lower_to_trt
-from fx2trt_oss.fx.utils import LowerPrecision
+from torch_tensorrt.fx.lower import lower_to_trt
+from torch_tensorrt.fx.utils import LowerPrecision
 
 
 """
diff --git a/py/torch_tensorrt/fx/example/quantized_resnet_test.py b/py/torch_tensorrt/fx/example/quantized_resnet_test.py
index 8b703a38cd..13a044b53b 100644
--- a/py/torch_tensorrt/fx/example/quantized_resnet_test.py
+++ b/py/torch_tensorrt/fx/example/quantized_resnet_test.py
@@ -1,16 +1,16 @@
 import copy
 
-import fx2trt_oss.tracer.acc_tracer.acc_tracer as acc_tracer
-
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch.fx
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_tracer as acc_tracer
 import torchvision.models as models
-from fx2trt_oss.fx import InputTensorSpec, TRTInterpreter, TRTModule
-from fx2trt_oss.fx.utils import LowerPrecision
 from torch.ao.quantization.quantize_fx import convert_fx, prepare_fx
 from torch.fx.experimental.normalize import NormalizeArgs
 from torch.fx.passes import shape_prop
+from torch_tensorrt.fx import InputTensorSpec, TRTInterpreter, TRTModule
+from torch_tensorrt.fx.utils import LowerPrecision
 
 rn18 = models.resnet18().eval()
 
diff --git a/py/torch_tensorrt/fx/example/test_fx2trt.py b/py/torch_tensorrt/fx/example/test_fx2trt.py
index 39225ce5e2..effc188e7a 100644
--- a/py/torch_tensorrt/fx/example/test_fx2trt.py
+++ b/py/torch_tensorrt/fx/example/test_fx2trt.py
@@ -1,29 +1,35 @@
-import torch_tensorrt
 import torch
+import torch_tensorrt
+
 
 class MyModel(torch.nn.Module):
     def __init__(self):
         super().__init__()
-        self.linear = torch.nn.Linear(5,3)
+        self.linear = torch.nn.Linear(5, 3)
         self.relu = torch.nn.functional.relu
-    def forward(self,x):
+
+    def forward(self, x):
         x = self.linear(x)
         x = self.relu(x)
         return x
 
 
-model = MyModel().eval() # torch module needs to be in eval (not training) mode
+model = MyModel().eval()  # torch module needs to be in eval (not training) mode
 
 # torch tensorrt
-inputs = [torch_tensorrt.Input(
-            (2,5),
-            dtype=torch.half,
-        )]
-enabled_precisions = {torch.float, torch.half} # Run with fp16
-
-trt_ts_module = torch_tensorrt.compile(model, inputs=inputs, enabled_precisions=enabled_precisions)
-
-inputs_ts = [torch.ones(2,5)]
+inputs = [
+    torch_tensorrt.Input(
+        (2, 5),
+        dtype=torch.half,
+    )
+]
+enabled_precisions = {torch.float, torch.half}  # Run with fp16
+
+trt_ts_module = torch_tensorrt.compile(
+    model, inputs=inputs, enabled_precisions=enabled_precisions
+)
+
+inputs_ts = [torch.ones(2, 5)]
 inputs_ts = [i.cuda().half() for i in inputs_ts]
 result = trt_ts_module(*inputs_ts)
 print(result)
@@ -33,12 +39,14 @@ def forward(self,x):
 print(ref)
 
 # fx2trt
-inputs_fx = [torch.ones((2,5))]
+inputs_fx = [torch.ones((2, 5))]
 
 model.cuda().half()
 inputs_fx = [i.cuda().half() for i in inputs_fx]
 
-trt_fx_module = torch_tensorrt.compile(model, ir="fx", inputs=inputs_fx, enabled_precisions={torch.half})
+trt_fx_module = torch_tensorrt.compile(
+    model, ir="fx", inputs=inputs_fx, enabled_precisions={torch.half}
+)
 result = trt_fx_module(*inputs_fx)
 print(result)
 
diff --git a/py/torch_tensorrt/fx/example/torchdynamo_example.py b/py/torch_tensorrt/fx/example/torchdynamo_example.py
index 79ba3bf1b3..6bb93f6d6e 100644
--- a/py/torch_tensorrt/fx/example/torchdynamo_example.py
+++ b/py/torch_tensorrt/fx/example/torchdynamo_example.py
@@ -5,8 +5,8 @@
 import torch
 import torchdynamo
 import torchvision
-from fx2trt_oss.fx.lower import lower_to_trt
-from fx2trt_oss.fx.utils import LowerPrecision
+from torch_tensorrt.fx.lower import lower_to_trt
+from torch_tensorrt.fx.utils import LowerPrecision
 from torchdynamo.optimizations import backends
 
 """
diff --git a/py/torch_tensorrt/fx/fx2trt.py b/py/torch_tensorrt/fx/fx2trt.py
index e1a573e664..16da30575f 100644
--- a/py/torch_tensorrt/fx/fx2trt.py
+++ b/py/torch_tensorrt/fx/fx2trt.py
@@ -9,12 +9,12 @@
 import tensorrt as trt
 import torch
 import torch.fx
-from .observer import Observer
 from torch.fx.node import _get_qualified_name
 from torch.fx.passes.shape_prop import TensorMetadata
 
 from .converter_registry import CONVERTERS
 from .input_tensor_spec import InputTensorSpec
+from .observer import Observer
 from .utils import get_dynamic_dims, LowerPrecision, torch_dtype_to_trt
 
 
diff --git a/py/torch_tensorrt/fx/lower.py b/py/torch_tensorrt/fx/lower.py
index 3706eae74b..601408fceb 100644
--- a/py/torch_tensorrt/fx/lower.py
+++ b/py/torch_tensorrt/fx/lower.py
@@ -2,23 +2,27 @@
 import logging
 from typing import Any, Callable, Sequence
 
-from .tracer.acc_tracer import acc_tracer
-
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
 import torch.fx as fx
 import torch.nn as nn
-from .lower_setting import LowerSetting
-from .passes.pass_utils import decorate_method, validate_inference
-from .passes.splitter_base import SplitResult
+from torch.fx.passes.splitter_base import SplitResult
 
 from .fx2trt import TRTInterpreter, TRTInterpreterResult
 from .input_tensor_spec import InputTensorSpec
+from .lower_setting import LowerSetting
 from .passes.lower_pass_manager_builder import LowerPassManagerBuilder
-from .passes.pass_utils import chain_passes, PassFunc
+from .passes.pass_utils import (
+    chain_passes,
+    decorate_method,
+    PassFunc,
+    validate_inference,
+)
 from .tools.timing_cache_utils import TimingCacheManager
 from .tools.trt_splitter import TRTSplitter, TRTSplitterSetting
+
+from .tracer.acc_tracer import acc_tracer
 from .trt_module import TRTModule
 from .utils import LowerPrecision
 
diff --git a/py/torch_tensorrt/fx/lower_setting.py b/py/torch_tensorrt/fx/lower_setting.py
index 783a3c4eef..d1e1ba9d9a 100644
--- a/py/torch_tensorrt/fx/lower_setting.py
+++ b/py/torch_tensorrt/fx/lower_setting.py
@@ -1,15 +1,13 @@
 import dataclasses as dc
 from typing import List, Optional, Sequence, Set, Type
 
-from .input_tensor_spec import InputTensorSpec
-from .passes.lower_basic_pass import (
-    fuse_permute_linear,
-    fuse_permute_matmul,
-)
-from .utils import LowerPrecision
 from torch import nn
 from torch.fx.passes.pass_manager import PassManager
 
+from .input_tensor_spec import InputTensorSpec
+from .passes.lower_basic_pass import fuse_permute_linear, fuse_permute_matmul
+from .utils import LowerPrecision
+
 
 @dc.dataclass
 class LowerSetting:
diff --git a/py/torch_tensorrt/fx/passes/lower_basic_pass.py b/py/torch_tensorrt/fx/passes/lower_basic_pass.py
index f7f639e382..6dc2e86f22 100644
--- a/py/torch_tensorrt/fx/passes/lower_basic_pass.py
+++ b/py/torch_tensorrt/fx/passes/lower_basic_pass.py
@@ -3,13 +3,15 @@
 import warnings
 from typing import Any
 
-from ..tracer.acc_tracer import acc_ops
 import torch
 import torch.fx
+from torch.fx.experimental.const_fold import split_const_subgraphs
+
 from ..observer import observable
-from .pass_utils import log_before_after, validate_inference
+
+from ..tracer.acc_tracer import acc_ops
 from ..tracer.acc_tracer.acc_utils import get_attr
-from torch.fx.experimental.const_fold import split_const_subgraphs
+from .pass_utils import log_before_after, validate_inference
 
 # Create an alias for module input type to avoid littering pyre-ignore for Any
 # throughout the file.
@@ -46,15 +48,15 @@ def fuse_sparse_matmul_add(gm: torch.fx.GraphModule, input: Input):
     def forward(self, x):
         a = self.a
         b = self.b
-        addmm_mm = fx2trt_oss.tracer.acc_tracer.acc_ops.matmul(input = a, other = b);  a = b = None
-        addmm_add = fx2trt_oss.tracer.acc_tracer.acc_ops.add(input = addmm_mm, other = x);  addmm_mm = x = None
+        addmm_mm = torch_tensorrt.fx.tracer.acc_tracer.acc_ops.matmul(input = a, other = b);  a = b = None
+        addmm_add = torch_tensorrt.fx.tracer.acc_tracer.acc_ops.add(input = addmm_mm, other = x);  addmm_mm = x = None
         return addmm_add
 
     After:
     def forward(self, x):
         a = self.a
         b = self.b
-        linear_1 = fx2trt_oss.tracer.acc_tracer.acc_ops.linear(input = a, weight = b, bias = x);  a = b = x = None
+        linear_1 = torch_tensorrt.fx.tracer.acc_tracer.acc_ops.linear(input = a, weight = b, bias = x);  a = b = x = None
         return linear_1
     """
     counter = 0
@@ -198,8 +200,8 @@ def fuse_permute_matmul(gm: torch.fx.GraphModule, input: Input):
 try:
     # @manual=//deeplearning/trt/python:py_tensorrt
     import tensorrt as trt
-    from fx2trt_oss.fx.converter_registry import tensorrt_converter
-    from fx2trt_oss.fx.converters.converter_utils import (
+    from torch_tensorrt.fx.converter_registry import tensorrt_converter
+    from torch_tensorrt.fx.converters.converter_utils import (
         add_binary_elementwise_layer,
         broadcast,
         get_trt_tensor,
diff --git a/py/torch_tensorrt/fx/passes/lower_pass_manager_builder.py b/py/torch_tensorrt/fx/passes/lower_pass_manager_builder.py
index 78298b0c1d..0fc3557069 100644
--- a/py/torch_tensorrt/fx/passes/lower_pass_manager_builder.py
+++ b/py/torch_tensorrt/fx/passes/lower_pass_manager_builder.py
@@ -2,16 +2,15 @@
 from typing import Any, Callable, NamedTuple, Sequence
 
 import torch
-from ..lower_setting import LowerSetting
-from ..observer import Observer
-from .passes.remove_duplicate_output_args import (
-    remove_duplicate_output_args,
-)
 from torch import nn
 from torch.fx.passes.pass_manager import inplace_wrapper, PassManager
 from torch.fx.passes.shape_prop import ShapeProp
 from torch.fx.passes.splitter_base import SplitResult
 
+from ..lower_setting import LowerSetting
+from ..observer import Observer
+from ..passes.remove_duplicate_output_args import remove_duplicate_output_args
+
 from .lower_basic_pass import run_const_fold
 
 Input = Sequence[Any]
@@ -23,7 +22,7 @@
 # List of observers. We can subscribe to them by calling its `add(callback)`
 # function from anywhere in code:
 #
-# >>> from fx2trt_oss.fx.lower import FUSE_PASSES_POST_OBSERVER
+# >>> from torch_tensorrt.fx.lower import FUSE_PASSES_POST_OBSERVER
 # >>> with FUSE_PASSES_POST_OBSERVER.add(print_module_and_input):
 # >>>     # print_module_and_input will be called right after the fuse passes
 # >>>     lower(module, sample_input)
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_adaptive_avgpool.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_adaptive_avgpool.py
index fa9f5187fa..8f053b9bc9 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_adaptive_avgpool.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_adaptive_avgpool.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_any.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_any.py
index 62adadd4a5..08e676dbcb 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_any.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_any.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_as_strided.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_as_strided.py
index 8dba126ded..50b5eaa392 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_as_strided.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_as_strided.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_avgpool.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_avgpool.py
index f8811b34ff..7fabb45ffd 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_avgpool.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_avgpool.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_batchnorm.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_batchnorm.py
index 0f2253dda1..65cdfb3a17 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_batchnorm.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_batchnorm.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_binary_ops.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_binary_ops.py
index 164784bbe2..b0bed3dfd7 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_binary_ops.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_binary_ops.py
@@ -1,8 +1,9 @@
 from typing import Callable
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_cat.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_cat.py
index 2c4077a17e..485fae5589 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_cat.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_cat.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_chunk.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_chunk.py
index 68d1c23a3a..39a86fc497 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_chunk.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_chunk.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_clamp.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_clamp.py
index bc2791542d..ad3c70156b 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_clamp.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_clamp.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_convolution.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_convolution.py
index 43cb352bfd..5706b1c2c5 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_convolution.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_convolution.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_dequantize.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_dequantize.py
index 45d86beee1..9d4a98e28c 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_dequantize.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_dequantize.py
@@ -1,9 +1,10 @@
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import tensorrt as trt
 import torch.fx
 import torch.nn as nn
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_einsum.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_einsum.py
index fd95633b79..84923e32fa 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_einsum.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_einsum.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_elu.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_elu.py
index f89cc2b9d6..de3c1d3400 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_elu.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_elu.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_embedding.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_embedding.py
index 40db2677f6..e0e6843bb2 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_embedding.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_embedding.py
@@ -1,12 +1,16 @@
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
 
-@unittest.skip("Current implementation is limited. All implementations in hf use int64. T113156424")
+
+@unittest.skip(
+    "Current implementation is limited. All implementations in hf use int64. T113156424"
+)
 class TestEmbeddingConverter(AccTestCase):
     @parameterized.expand(
         [
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_eq.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_eq.py
index 0f5a6c5967..c459a382b0 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_eq.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_eq.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_expand.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_expand.py
index b71d6850cc..53143102aa 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_expand.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_expand.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_flatten.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_flatten.py
index d4572cbd56..f9ce31b3cc 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_flatten.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_flatten.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_gelu.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_gelu.py
index d022cce2ff..fbe415d1e3 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_gelu.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_gelu.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_getitem.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_getitem.py
index a81526eec8..547caa4866 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_getitem.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_getitem.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_gt.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_gt.py
index 34228956dd..c1e01295e4 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_gt.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_gt.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_hard_sigmoid.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_hard_sigmoid.py
index 8a329626e7..6e41caf279 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_hard_sigmoid.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_hard_sigmoid.py
@@ -1,9 +1,9 @@
 import torch
-from fx2trt_oss.tracer.acc_tracer import acc_ops
 from parameterized import parameterized
 from torch import nn
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.tracer.acc_tracer import acc_ops
 
 
 class TestHardSigmoid(AccTestCase):
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_hardtanh.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_hardtanh.py
index f84ec3aa20..6c1c56796a 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_hardtanh.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_hardtanh.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py
index 36060b5c80..4046e82bae 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_isinf.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_isinf.py
index 5f4e897d2f..21ab076c9b 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_isinf.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_isinf.py
@@ -1,7 +1,8 @@
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_leaky_relu.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_leaky_relu.py
index 8bdbf825df..7347796e23 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_leaky_relu.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_leaky_relu.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_linear.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_linear.py
index e5a37996cc..9f369de543 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_linear.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_linear.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_and.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_and.py
index e22d80091c..51915127b5 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_and.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_and.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_or.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_or.py
index 6c89816a93..49599fb181 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_or.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_or.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_xor.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_xor.py
index a7169b237a..44ab745dd5 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_xor.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_logical_xor.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_lt.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_lt.py
index 6e41a8dba6..4e1490ce52 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_lt.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_lt.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_masked_fill.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_masked_fill.py
index 88195c6703..aa105f512c 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_masked_fill.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_masked_fill.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_matmul.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_matmul.py
index 99fae5fcf2..f6ed85c156 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_matmul.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_matmul.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_max.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_max.py
index 79df23de4c..45c9224d50 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_max.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_max.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_maximum.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_maximum.py
index 8c206516a9..1fb18319c9 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_maximum.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_maximum.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_maxpool.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_maxpool.py
index 01ffa0c7cd..274fe27d12 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_maxpool.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_maxpool.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_min.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_min.py
index 08329429d5..317f71d3c0 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_min.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_min.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_minimum.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_minimum.py
index a66ebbebcf..0e9af3f302 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_minimum.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_minimum.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_narrow.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_narrow.py
index 995f32de1c..8422a88ad7 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_narrow.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_narrow.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_ne.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_ne.py
index fc659e632b..37b2432e72 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_ne.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_ne.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_new_ones.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_new_ones.py
index c3ae338e86..2ff64f55c2 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_new_ones.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_new_ones.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_numel.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_numel.py
index 7760b8a4e7..8b30dc9eb9 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_numel.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_numel.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_pad.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_pad.py
index f8e7df178d..edfd849790 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_pad.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_pad.py
@@ -1,9 +1,10 @@
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import tensorrt as trt
 import torch
 import torch.nn as nn
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_permute.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_permute.py
index b9223a4fa3..2d8d511749 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_permute.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_permute.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_prod.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_prod.py
index 2355fe18f2..9f112a8a91 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_prod.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_prod.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_quantize_per_tensor.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_quantize_per_tensor.py
index f4bfa06e7b..c7084e17fd 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_quantize_per_tensor.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_quantize_per_tensor.py
@@ -1,9 +1,10 @@
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import tensorrt as trt
 import torch.fx
 import torch.nn as nn
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_reduce_ops.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_reduce_ops.py
index 44aa01ed99..1d24d332d4 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_reduce_ops.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_reduce_ops.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_relu.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_relu.py
index 07a809930a..5757e1a138 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_relu.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_relu.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_repeat_interleave.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_repeat_interleave.py
index c8b94b3dab..6dc6d0d108 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_repeat_interleave.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_repeat_interleave.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch import nn
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_reshape.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_reshape.py
index 85a404b55d..fd62c4cebd 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_reshape.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_reshape.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_selu.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_selu.py
index b8e285cd79..ddb953abb5 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_selu.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_selu.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_sigmoid.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_sigmoid.py
index ad09892afa..32ba1009ca 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_sigmoid.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_sigmoid.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_silu.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_silu.py
index 72dc1f84d4..cc9f32e4d5 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_silu.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_silu.py
@@ -1,8 +1,8 @@
 import torch
-from fx2trt_oss.tracer.acc_tracer import acc_ops
 from torch import nn
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.tracer.acc_tracer import acc_ops
 
 
 class TestSilu(AccTestCase):
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_size.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_size.py
index 7faf544d01..dc128f5b68 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_size.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_size.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_softmax.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_softmax.py
index 09a40cef90..c6ec1cc578 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_softmax.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_softmax.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_softsign.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_softsign.py
index ec20a00cbf..d64c8838c4 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_softsign.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_softsign.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_split.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_split.py
index d3bf20e375..c71b1b8d65 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_split.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_split.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_squeeze.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_squeeze.py
index 3d16dfbe59..9460440ee1 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_squeeze.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_squeeze.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_std.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_std.py
index 29096f00b7..96cf6167ed 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_std.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_std.py
@@ -1,5 +1,5 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_tanh.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_tanh.py
index 92450346bc..7dd611e9fd 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_tanh.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_tanh.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
 
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_tile.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_tile.py
index 6566df84d5..d303f50f2c 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_tile.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_tile.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_to_dtype.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_to_dtype.py
index 65459f84c5..09c321c136 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_to_dtype.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_to_dtype.py
@@ -1,8 +1,8 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
-from fx2trt_oss.fx.utils import LowerPrecision
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.utils import LowerPrecision
 
 
 class TestToConverter(AccTestCase):
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_topk.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_topk.py
index 13d39b86be..d8ea1ef396 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_topk.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_topk.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_transpose_convolution.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_transpose_convolution.py
index 7fed27b944..e9d408714e 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_transpose_convolution.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_transpose_convolution.py
@@ -1,7 +1,7 @@
 # Owner(s): ["oncall: gpu_enablement"]
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_type_as.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_type_as.py
index 0f2cb5438f..c065550f51 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_type_as.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_type_as.py
@@ -1,8 +1,8 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
-from fx2trt_oss.fx.utils import LowerPrecision
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.utils import LowerPrecision
 
 
 class TestTypeAsConverter(AccTestCase):
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_unary_ops.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_unary_ops.py
index f9af56835e..d11fb269bf 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_unary_ops.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_unary_ops.py
@@ -1,8 +1,9 @@
 from typing import Callable
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_unsqueeze.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_unsqueeze.py
index 79064771b7..e31b4f2481 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_unsqueeze.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_unsqueeze.py
@@ -1,7 +1,7 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.fx
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_where.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_where.py
index 3b4a33a891..c90bb16041 100644
--- a/py/torch_tensorrt/fx/test/converters/acc_op/test_where.py
+++ b/py/torch_tensorrt/fx/test/converters/acc_op/test_where.py
@@ -1,6 +1,6 @@
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
 import torch.nn as nn
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
diff --git a/py/torch_tensorrt/fx/test/core/test_import_fx2trt.py b/py/torch_tensorrt/fx/test/core/test_import_fx2trt.py
index acf300c3d7..7f8dd169be 100644
--- a/py/torch_tensorrt/fx/test/core/test_import_fx2trt.py
+++ b/py/torch_tensorrt/fx/test/core/test_import_fx2trt.py
@@ -5,7 +5,7 @@
 
 # Test that this import should not trigger any error when run
 # in non-GPU hosts, or in any build mode.
-import fx2trt_oss.fx.lower as fxl  # noqa: F401
+import torch_tensorrt.fx.lower as fxl  # noqa: F401
 from torch.testing._internal.common_utils import run_tests, TestCase
 
 
diff --git a/py/torch_tensorrt/fx/test/core/test_input_tensor_spec.py b/py/torch_tensorrt/fx/test/core/test_input_tensor_spec.py
index ee57bbf2bd..cec49cb400 100644
--- a/py/torch_tensorrt/fx/test/core/test_input_tensor_spec.py
+++ b/py/torch_tensorrt/fx/test/core/test_input_tensor_spec.py
@@ -3,8 +3,8 @@
 from typing import List, Optional
 
 import torch
-from fx2trt_oss.fx import InputTensorSpec
 from torch.testing._internal.common_utils import run_tests, TestCase
+from torch_tensorrt.fx import InputTensorSpec
 
 
 class TestTRTModule(TestCase):
diff --git a/py/torch_tensorrt/fx/test/core/test_trt_module.py b/py/torch_tensorrt/fx/test/core/test_trt_module.py
index 2a1eaa269e..ce44be758d 100644
--- a/py/torch_tensorrt/fx/test/core/test_trt_module.py
+++ b/py/torch_tensorrt/fx/test/core/test_trt_module.py
@@ -2,12 +2,13 @@
 
 import os
 
-import fx2trt_oss.tracer.acc_tracer.acc_tracer as acc_tracer
 import torch
 import torch.fx
-from fx2trt_oss.fx import InputTensorSpec, TRTInterpreter, TRTModule
-from fx2trt_oss.fx.utils import LowerPrecision
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_tracer as acc_tracer
 from torch.testing._internal.common_utils import run_tests, TestCase
+from torch_tensorrt.fx import InputTensorSpec, TRTInterpreter, TRTModule
+from torch_tensorrt.fx.utils import LowerPrecision
 
 
 class TestTRTModule(TestCase):
diff --git a/py/torch_tensorrt/fx/test/passes/test_fuse_permute_linear_trt.py b/py/torch_tensorrt/fx/test/passes/test_fuse_permute_linear_trt.py
index 4822b87af6..6eca458ff5 100644
--- a/py/torch_tensorrt/fx/test/passes/test_fuse_permute_linear_trt.py
+++ b/py/torch_tensorrt/fx/test/passes/test_fuse_permute_linear_trt.py
@@ -1,13 +1,13 @@
 # Owner(s): ["oncall: gpu_enablement"]
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
-from fx2trt_oss.fx.passes.lower_basic_pass import (
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
+from torch.testing._internal.common_fx2trt import AccTestCase
+from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.passes.lower_basic_pass import (
     fuse_permute_linear,
     trt_transposed_linear,
 )
-from torch.testing._internal.common_fx2trt import AccTestCase
-from torch.testing._internal.common_utils import run_tests
 
 
 class TestFusePermuteLinear(AccTestCase):
diff --git a/py/torch_tensorrt/fx/test/passes/test_fuse_permute_matmul_trt.py b/py/torch_tensorrt/fx/test/passes/test_fuse_permute_matmul_trt.py
index 4b00fb43e6..dab797e02b 100644
--- a/py/torch_tensorrt/fx/test/passes/test_fuse_permute_matmul_trt.py
+++ b/py/torch_tensorrt/fx/test/passes/test_fuse_permute_matmul_trt.py
@@ -1,14 +1,14 @@
 # Owner(s): ["oncall: gpu_enablement"]
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
-from fx2trt_oss.fx.passes.lower_basic_pass import (
-    fuse_permute_matmul,
-    trt_transposed_matmul,
-)
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from parameterized import param, parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.passes.lower_basic_pass import (
+    fuse_permute_matmul,
+    trt_transposed_matmul,
+)
 
 
 def tranpose_last_two_dims(x):
diff --git a/py/torch_tensorrt/fx/test/passes/test_multi_fuse_trt.py b/py/torch_tensorrt/fx/test/passes/test_multi_fuse_trt.py
index db9318ca3f..37e7b17b01 100644
--- a/py/torch_tensorrt/fx/test/passes/test_multi_fuse_trt.py
+++ b/py/torch_tensorrt/fx/test/passes/test_multi_fuse_trt.py
@@ -1,16 +1,16 @@
 # Owner(s): ["oncall: gpu_enablement"]
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch
-from fx2trt_oss.fx.passes.lower_basic_pass import (
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
+from parameterized import parameterized
+from torch.testing._internal.common_fx2trt import AccTestCase
+from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.passes.lower_basic_pass import (
     fuse_permute_linear,
     fuse_permute_matmul,
     trt_transposed_linear,
     trt_transposed_matmul,
 )
-from parameterized import parameterized
-from torch.testing._internal.common_fx2trt import AccTestCase
-from torch.testing._internal.common_utils import run_tests
 
 
 def permute021(x):
diff --git a/py/torch_tensorrt/fx/test/passes/test_remove_duplicate_output_args.py b/py/torch_tensorrt/fx/test/passes/test_remove_duplicate_output_args.py
index 24977d49df..f20598f12e 100644
--- a/py/torch_tensorrt/fx/test/passes/test_remove_duplicate_output_args.py
+++ b/py/torch_tensorrt/fx/test/passes/test_remove_duplicate_output_args.py
@@ -2,9 +2,10 @@
 
 import logging
 
-import fx2trt_oss.fx.passes.remove_duplicate_output_args as dedup
 import torch.fx as fx
 import torch.nn as nn
+
+import torch_tensorrt.fx.passes.remove_duplicate_output_args as dedup
 from torch.testing._internal.common_utils import run_tests, TestCase
 
 
diff --git a/py/torch_tensorrt/fx/test/passes/test_setitem.py b/py/torch_tensorrt/fx/test/passes/test_setitem.py
index 61058d07bf..ddb6d5255b 100644
--- a/py/torch_tensorrt/fx/test/passes/test_setitem.py
+++ b/py/torch_tensorrt/fx/test/passes/test_setitem.py
@@ -1,9 +1,9 @@
 import torch
 import torchdynamo
-from fx2trt_oss.fx.passes.lower_basic_pass import transform_setitem
 from parameterized import parameterized
 from torch.testing._internal.common_fx2trt import AccTestCase
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx.passes.lower_basic_pass import transform_setitem
 from torchdynamo.optimizations import backends
 
 
diff --git a/py/torch_tensorrt/fx/test/quant/test_quant_trt.py b/py/torch_tensorrt/fx/test/quant/test_quant_trt.py
index 3f9cc99817..1f93a9f397 100644
--- a/py/torch_tensorrt/fx/test/quant/test_quant_trt.py
+++ b/py/torch_tensorrt/fx/test/quant/test_quant_trt.py
@@ -5,15 +5,12 @@
 import operator
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_tracer as acc_tracer
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.nn.quantized._reference as nnqr
-from fx2trt_oss.fx import InputTensorSpec, TRTInterpreter, TRTModule
-from fx2trt_oss.fx.passes.lower_basic_pass import run_const_fold
-from fx2trt_oss.fx.utils import LowerPrecision
-from fx2trt_oss.tracer.acc_tracer import acc_ops
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_tracer as acc_tracer
 from torch.ao.quantization import default_qconfig
 from torch.ao.quantization.backend_config.observation_type import ObservationType
 from torch.ao.quantization.fx.match_utils import MatchAllNode
@@ -29,6 +26,10 @@
     QuantizationTestCase,
 )
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt.fx import InputTensorSpec, TRTInterpreter, TRTModule
+from torch_tensorrt.fx.passes.lower_basic_pass import run_const_fold
+from torch_tensorrt.fx.tracer.acc_tracer import acc_ops
+from torch_tensorrt.fx.utils import LowerPrecision
 
 
 def lower_to_trt(model, inputs, shape_ranges):
diff --git a/py/torch_tensorrt/fx/test/tools/test_model_packager.py b/py/torch_tensorrt/fx/test/tools/test_model_packager.py
index 063dd22b40..b0ef521d27 100644
--- a/py/torch_tensorrt/fx/test/tools/test_model_packager.py
+++ b/py/torch_tensorrt/fx/test/tools/test_model_packager.py
@@ -3,9 +3,12 @@
 
 import torch
 import torch.fx
-from fx2trt_oss.fx.tools.model_packager import generate_standalone_repro, ModelPackager
 from torch import nn
 from torch.package import PackageImporter
+from torch_tensorrt.fx.tools.model_packager import (
+    generate_standalone_repro,
+    ModelPackager,
+)
 
 
 class TestModel(nn.Module):
diff --git a/py/torch_tensorrt/fx/test/tracer/test_acc_shape_prop.py b/py/torch_tensorrt/fx/test/tracer/test_acc_shape_prop.py
index da8ed86190..a2f842b722 100644
--- a/py/torch_tensorrt/fx/test/tracer/test_acc_shape_prop.py
+++ b/py/torch_tensorrt/fx/test/tracer/test_acc_shape_prop.py
@@ -3,9 +3,10 @@
 import operator
 import unittest
 
-import fx2trt_oss.tracer.acc_tracer.acc_shape_prop as acc_shape_prop
-import fx2trt_oss.tracer.acc_tracer.acc_tracer as acc_tracer
 import torch
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_shape_prop as acc_shape_prop
+import torch_tensorrt.fx.tracer.acc_tracer.acc_tracer as acc_tracer
 from parameterized import param, parameterized
 
 torch.manual_seed(0)
diff --git a/py/torch_tensorrt/fx/test/tracer/test_acc_tracer.py b/py/torch_tensorrt/fx/test/tracer/test_acc_tracer.py
index 0f10286072..231b8eed0c 100644
--- a/py/torch_tensorrt/fx/test/tracer/test_acc_tracer.py
+++ b/py/torch_tensorrt/fx/test/tracer/test_acc_tracer.py
@@ -3,13 +3,14 @@
 import unittest
 from typing import Callable, List
 
-import fx2trt_oss.tracer.acc_tracer.acc_normalizer as acc_normalizer
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
-import fx2trt_oss.tracer.acc_tracer.acc_tracer as acc_tracer
-import fx2trt_oss.tracer.acc_tracer.acc_utils as acc_utils
 import numpy as np
 import torch
 import torch.nn as nn
+
+import torch_tensorrt.fx.tracer.acc_tracer.acc_normalizer as acc_normalizer
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
+import torch_tensorrt.fx.tracer.acc_tracer.acc_tracer as acc_tracer
+import torch_tensorrt.fx.tracer.acc_tracer.acc_utils as acc_utils
 import torchvision
 from parameterized import param, parameterized
 
@@ -1725,9 +1726,10 @@ def test_trunc_div(self):
         self._make_acc_op_function_test(
             acc_ops.trunc_div, lambda x: torch.div(x, 2, rounding_mode="trunc")
         )
-        self._make_acc_op_function_test(
-            acc_ops.trunc_div, lambda x: torch.floor_divide(x, 2)
-        )
+        # does not behave the same as floor_divide
+        # self._make_acc_op_function_test(
+        #     acc_ops.trunc_div, lambda x: torch.floor_divide(x, 2)
+        # )
 
     def test_view(self):
         """
diff --git a/py/torch_tensorrt/fx/test/tracer/test_dispatch_tracer.py b/py/torch_tensorrt/fx/test/tracer/test_dispatch_tracer.py
index fea3cf6099..c5b7a22ec6 100644
--- a/py/torch_tensorrt/fx/test/tracer/test_dispatch_tracer.py
+++ b/py/torch_tensorrt/fx/test/tracer/test_dispatch_tracer.py
@@ -6,8 +6,8 @@
 
 from functorch import make_fx as make_fx_pk
 from functorch.experimental import functionalize
-from fx2trt_oss.tracer.dispatch_tracer.tracer import make_fx
 from torch.library import Library
+from torch_tensorrt.fx.tracer.dispatch_tracer.tracer import make_fx
 from torchdynamo.optimizations.normalize import normalize_ir
 from torchdynamo.optimizations.python_key import fake_signature
 
diff --git a/py/torch_tensorrt/fx/test/trt_lower/test_diagnostics.py b/py/torch_tensorrt/fx/test/trt_lower/test_diagnostics.py
index e62cd62bca..467f3ca9af 100644
--- a/py/torch_tensorrt/fx/test/trt_lower/test_diagnostics.py
+++ b/py/torch_tensorrt/fx/test/trt_lower/test_diagnostics.py
@@ -7,7 +7,7 @@
 from typing import Union
 from unittest import TestCase
 
-import fx2trt_oss.fx.diagnostics as diag
+import torch_tensorrt.fx.diagnostics as diag
 
 
 def reset_diag(fn):
diff --git a/py/torch_tensorrt/fx/test/trt_lower/test_fx2trt_lower.py b/py/torch_tensorrt/fx/test/trt_lower/test_fx2trt_lower.py
index c4d6346c35..7ecfff69c8 100644
--- a/py/torch_tensorrt/fx/test/trt_lower/test_fx2trt_lower.py
+++ b/py/torch_tensorrt/fx/test/trt_lower/test_fx2trt_lower.py
@@ -6,7 +6,7 @@
 import torch
 import torch.fx as fx
 import torch.nn as nn
-from fx2trt_oss.fx.lower import Lowerer, LowerSetting
+from torch_tensorrt.fx.lower import Lowerer, LowerSetting
 
 logger = logging.getLogger(__name__)
 
diff --git a/py/torch_tensorrt/fx/test/trt_lower/test_observer.py b/py/torch_tensorrt/fx/test/trt_lower/test_observer.py
index f5ca2b3ba7..8a621c476a 100644
--- a/py/torch_tensorrt/fx/test/trt_lower/test_observer.py
+++ b/py/torch_tensorrt/fx/test/trt_lower/test_observer.py
@@ -4,8 +4,8 @@
 from contextlib import contextmanager
 from unittest import TestCase
 
-import fx2trt_oss.fx.observer as ob
-from fx2trt_oss.fx.observer import observable
+import torch_tensorrt.fx.observer as ob
+from torch_tensorrt.fx.observer import observable
 
 
 def set_observer_callback_rethrow(fn):
diff --git a/py/torch_tensorrt/fx/test/trt_lower/test_observer_gpu.py b/py/torch_tensorrt/fx/test/trt_lower/test_observer_gpu.py
index 26934c9f06..9ed3b9df06 100644
--- a/py/torch_tensorrt/fx/test/trt_lower/test_observer_gpu.py
+++ b/py/torch_tensorrt/fx/test/trt_lower/test_observer_gpu.py
@@ -2,9 +2,9 @@
 import functools
 from unittest import TestCase
 
-import fx2trt_oss.fx.observer as ob
-from fx2trt_oss.fx.passes.lower_basic_pass import fuse_permute_linear
+import torch_tensorrt.fx.observer as ob
 from test_observer import execution_verifier, set_observer_callback_rethrow
+from torch_tensorrt.fx.passes.lower_basic_pass import fuse_permute_linear
 
 
 class ObserverGPUTests(TestCase):
@@ -16,10 +16,11 @@ def test_observe_lowerer(self):
         """
         from dataclasses import replace
 
-        import fx2trt_oss.fx.lower as lower
         import torch
         import torch.nn as nn
-        from fx2trt_oss.fx.lower_setting import LowerSetting
+
+        import torch_tensorrt.fx.lower as lower
+        from torch_tensorrt.fx.lower_setting import LowerSetting
 
         class Model(nn.Module):
             def forward(self, x, y):
diff --git a/py/torch_tensorrt/fx/test/trt_lower/trt_operator_supported_test.py b/py/torch_tensorrt/fx/test/trt_lower/trt_operator_supported_test.py
index 9c74579909..0942909c94 100644
--- a/py/torch_tensorrt/fx/test/trt_lower/trt_operator_supported_test.py
+++ b/py/torch_tensorrt/fx/test/trt_lower/trt_operator_supported_test.py
@@ -1,12 +1,12 @@
 # Owner(s): ["oncall: gpu_enablement"]
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops  # noqa: F401
 import torch
 import torch.fx
 import torch.nn as nn
-from fx2trt_oss.fx.tools.trt_splitter import create_trt_operator_support
-from fx2trt_oss.tracer.acc_tracer import acc_ops, acc_tracer
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops  # noqa: F401
 from torch.testing._internal.common_utils import run_tests, TestCase
+from torch_tensorrt.fx.tools.trt_splitter import create_trt_operator_support
+from torch_tensorrt.fx.tracer.acc_tracer import acc_ops, acc_tracer
 
 
 class TestTRTOperatorSupport(TestCase):
diff --git a/py/torch_tensorrt/fx/test/trt_lower/trt_splitter_test.py b/py/torch_tensorrt/fx/test/trt_lower/trt_splitter_test.py
index 917afea0f2..76da8c2430 100644
--- a/py/torch_tensorrt/fx/test/trt_lower/trt_splitter_test.py
+++ b/py/torch_tensorrt/fx/test/trt_lower/trt_splitter_test.py
@@ -5,13 +5,13 @@
 import torch  # isort:skip
 import torch.fx  # isort:skip
 
-import fx2trt_oss.tracer.acc_tracer.acc_ops as acc_ops
 import torch.fx.passes.operator_support as op_support
 import torch.fx.passes.shape_prop as shape_prop
-from fx2trt_oss.fx.tools.trt_splitter import TRTSplitter
-from fx2trt_oss.tracer.acc_tracer import acc_tracer
+import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
 from torch.fx.passes import splitter_base
 from torch.testing._internal.common_utils import run_tests, TestCase
+from torch_tensorrt.fx.tools.trt_splitter import TRTSplitter
+from torch_tensorrt.fx.tracer.acc_tracer import acc_tracer
 
 
 ERROR_MSG_NO_ACC_MODULE = "FX split failed: Did not find any ACC submodule!"
diff --git a/py/torch_tensorrt/fx/tools/engine_layer_visualize.py b/py/torch_tensorrt/fx/tools/engine_layer_visualize.py
index eaff4aafc9..3008c8e087 100644
--- a/py/torch_tensorrt/fx/tools/engine_layer_visualize.py
+++ b/py/torch_tensorrt/fx/tools/engine_layer_visualize.py
@@ -14,7 +14,7 @@
 supported.
 
 Usage:
-    python fx2trt_oss.fx/tools/engine_layer_visualize.py --log_file aaa --profile_file bbb
+    python torch_tensorrt.fx/tools/engine_layer_visualize.py --log_file aaa --profile_file bbb
 """
 
 
diff --git a/py/torch_tensorrt/fx/tools/model_packager.py b/py/torch_tensorrt/fx/tools/model_packager.py
index fd2645c108..0ef0ff05a4 100644
--- a/py/torch_tensorrt/fx/tools/model_packager.py
+++ b/py/torch_tensorrt/fx/tools/model_packager.py
@@ -102,9 +102,9 @@ def set_mocked_modules(cls, pe: PackageExporter):
         pe.mock(
             "**",
             exclude=[
-                "fx2trt_oss.tracer.acc_tracer.acc_ops",
-                "fx2trt_oss.tracer.acc_tracer.acc_normalizer",
-                "fx2trt_oss.tracer.acc_tracer.acc_op_properties",
+                "torch_tensorrt.fx.tracer.acc_tracer.acc_ops",
+                "torch_tensorrt.fx.tracer.acc_tracer.acc_normalizer",
+                "torch_tensorrt.fx.tracer.acc_tracer.acc_op_properties",
             ],
         )
 
diff --git a/py/torch_tensorrt/fx/tools/trt_minimizer.py b/py/torch_tensorrt/fx/tools/trt_minimizer.py
index 3aaefac442..23039bea34 100644
--- a/py/torch_tensorrt/fx/tools/trt_minimizer.py
+++ b/py/torch_tensorrt/fx/tools/trt_minimizer.py
@@ -2,9 +2,10 @@
 
 import torch
 import torch.fx.passes.net_min_base as net_min_base
-from .. import InputTensorSpec, TRTInterpreter, TRTModule
 from torch.fx.passes.tools_common import Tensors
 
+from .. import InputTensorSpec, TRTInterpreter, TRTModule
+
 
 def lower_mod_default(
     mod: torch.fx.GraphModule, inputs: Tensors, batch_size: Any = 2048
diff --git a/py/torch_tensorrt/fx/tools/trt_profiler_sorted.py b/py/torch_tensorrt/fx/tools/trt_profiler_sorted.py
index 529bd2fd70..2d5ce0b419 100644
--- a/py/torch_tensorrt/fx/tools/trt_profiler_sorted.py
+++ b/py/torch_tensorrt/fx/tools/trt_profiler_sorted.py
@@ -4,6 +4,7 @@
 
 import tensorrt as trt
 import torch
+
 from .. import TRTModule
 
 
diff --git a/py/torch_tensorrt/fx/tools/trt_splitter.py b/py/torch_tensorrt/fx/tools/trt_splitter.py
index c3efe74cc7..32dc4a1853 100644
--- a/py/torch_tensorrt/fx/tools/trt_splitter.py
+++ b/py/torch_tensorrt/fx/tools/trt_splitter.py
@@ -3,6 +3,8 @@
 import torch
 import torch.fx.passes.operator_support as ops
 import torch.fx.passes.splitter_base as splitter_base
+from torch.fx.passes.tools_common import get_acc_ops_name, Tensors
+
 from .. import (
     CONVERTERS,
     InputTensorSpec,
@@ -12,7 +14,6 @@
     TRTModule,
 )
 from ..tools.trt_minimizer import TensorRTMinimizer
-from torch.fx.passes.tools_common import get_acc_ops_name, Tensors
 
 
 def create_trt_operator_support(use_implicit_batch_dim=True) -> ops.OperatorSupportBase:
diff --git a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_normalizer.py b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_normalizer.py
index 89cb299e0d..408744726e 100644
--- a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_normalizer.py
+++ b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_normalizer.py
@@ -2,11 +2,12 @@
 import re
 from typing import Any, Callable, Dict, List, NamedTuple, Optional, Set, Tuple, Union
 
-from . import acc_utils
 import torch
 import torch.fx
 from torch.fx.node import _get_qualified_name
 
+from . import acc_utils
+
 # Need to keep up-to-date with https://fburl.com/codesearch/7r2hhh53
 ALIAS_MAP = {
     "input": ("input", "x", "a", "x1"),
diff --git a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py
index 3f2d18d089..df6480166b 100644
--- a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py
+++ b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py
@@ -5,18 +5,16 @@
 import torch  # isort:skip
 from typing import cast, Iterable, List, Sequence
 
-from . import acc_utils
 import torch.nn as nn
+from torch.fx.passes.shape_prop import _extract_tensor_metadata
+
+from . import acc_utils
 from .acc_normalizer import (
     register_acc_op,
     register_acc_op_mapping,
     register_custom_acc_mapper_fn,
 )
-from .acc_op_properties import (
-    AccOpProperty,
-    register_acc_op_properties,
-)
-from torch.fx.passes.shape_prop import _extract_tensor_metadata
+from .acc_op_properties import AccOpProperty, register_acc_op_properties
 
 this_arg_is_optional = True
 move_to_qparams = True
@@ -223,6 +221,7 @@ def avg_pool2d(
 def sign(*, input):
     return torch.sign(input)
 
+
 @register_custom_acc_mapper_fn(
     op_and_target=("call_method", "type"),
     arg_replacement_tuples=[
diff --git a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_tracer.py b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_tracer.py
index 1a8b6499b0..9f3f911261 100644
--- a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_tracer.py
+++ b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_tracer.py
@@ -8,10 +8,6 @@
 from types import FunctionType
 from typing import Any, Dict, Optional, Sequence, Set, Tuple, Type
 
-from . import acc_normalizer
-from . import acc_ops  # noqa: F401
-from . import acc_shape_prop
-from . import acc_utils
 import torch
 import torch.jit as jit
 import torch.nn as nn
@@ -21,6 +17,8 @@
 from torch.fx.node import Argument, Node, Target
 from torch.fx.passes import shape_prop
 
+from . import acc_normalizer, acc_ops, acc_shape_prop, acc_utils  # noqa: F401
+
 
 _LOGGER = logging.getLogger(__name__)