From 5557c1d642789da7cf4886e11cb5d58f7af3bff4 Mon Sep 17 00:00:00 2001 From: Ruturaj Vaidya Date: Thu, 10 Apr 2025 10:33:25 -0500 Subject: [PATCH] Fix circular import in pallas core file (#354) --- jax/_src/pallas/mosaic/core.py | 1 - 1 file changed, 1 deletion(-) diff --git a/jax/_src/pallas/mosaic/core.py b/jax/_src/pallas/mosaic/core.py index 82fe9c2ba..bf6899fbe 100644 --- a/jax/_src/pallas/mosaic/core.py +++ b/jax/_src/pallas/mosaic/core.py @@ -28,7 +28,6 @@ from jax._src import core as jax_core from jax._src import dtypes from jax._src import util from jax._src.pallas import core as pallas_core -from jax._src.pallas import pallas_call import jax.numpy as jnp import numpy as np