Skip to content

Commit

Permalink
Removed unused imports in jax.experimental.mosaic.gpu.core
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 679289036
  • Loading branch information
superbobry authored and Google-ML-Automation committed Sep 27, 2024
1 parent ab4590c commit a8a9671
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions jax/experimental/mosaic/gpu/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,21 +19,15 @@
import dataclasses
import functools
import hashlib
import itertools
import math
import os
import pathlib
import subprocess
import tempfile
import time
from typing import Any, Generic, TypeVar
import weakref

import jax
from jax._src import config
from jax._src import core as jax_core
from jax._src.interpreters import mlir
from jax._src.lib import xla_client
from jaxlib.mlir import ir
from jaxlib.mlir.dialects import arith
from jaxlib.mlir.dialects import builtin
Expand All @@ -42,7 +36,6 @@
from jaxlib.mlir.dialects import llvm
from jaxlib.mlir.dialects import memref
from jaxlib.mlir.dialects import nvvm
from jaxlib.mlir.passmanager import PassManager
import numpy as np

from . import profiler
Expand Down

0 comments on commit a8a9671

Please sign in to comment.