diff --git a/crossfit/backend/torch/op/base.py b/crossfit/backend/torch/op/base.py
index 6330b92..829208d 100644
--- a/crossfit/backend/torch/op/base.py
+++ b/crossfit/backend/torch/op/base.py
@@ -13,11 +13,11 @@
 # limitations under the License.
 
 import gc
+from typing import Optional
 
 import cudf
 import cupy as cp
 import torch
-from typing import Optional
 
 from crossfit.backend.cudf.series import (
     create_list_series_from_2d_ar,
diff --git a/crossfit/backend/torch/op/embed.py b/crossfit/backend/torch/op/embed.py
index 309a3de..b7a3d97 100644
--- a/crossfit/backend/torch/op/embed.py
+++ b/crossfit/backend/torch/op/embed.py
@@ -12,14 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import gc
-
-import cudf
-import cupy as cp
-import torch
-
-from crossfit.backend.cudf.series import create_list_series_from_2d_ar
-from crossfit.backend.torch.loader import DEFAULT_BATCH_SIZE, InMemoryLoader, SortedSeqLoader
+from crossfit.backend.torch.loader import DEFAULT_BATCH_SIZE
 from crossfit.backend.torch.model import Model
 from crossfit.backend.torch.op.base import Predictor