facing errors in the simple python script
PS C:\Users\mandar\Desktop\AMS\vtt\New folder> & 'c:\Users\mandar\AppData\Local\Programs\Python\Python310\python.exe' 'c:\Users\mandar.cursor\extensions\ms-python.debugpy-2025.6.0-win32-x64\bundled\libs\debugpy\launcher' '60370' '--' 'c:\Users\mandar\Desktop\AMS\vtt\New folder\test.py'
c:\Users\mandar\AppData\Local\Programs\Python\Python310\lib\site-packages\transformers\utils\generic.py:311: FutureWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
torch.utils._pytree.register_pytree_node(
Traceback (most recent call last):
File "c:\Users\mandar\Desktop\AMS\vtt\New folder\test.py", line 1, in
from whisper_jax import FlaxWhisperForConditionalGeneration, FlaxWhisperPipline
File "c:\Users\mandar\AppData\Local\Programs\Python\Python310\lib\site-packages\whisper_jax_init.py", line 18, in
from .modeling_flax_whisper import FlaxWhisperForConditionalGeneration
File "c:\Users\mandar\AppData\Local\Programs\Python\Python310\lib\site-packages\whisper_jax\modeling_flax_whisper.py", line 57, in
from whisper_jax import layers
File "c:\Users\mandar\AppData\Local\Programs\Python\Python310\lib\site-packages\whisper_jax\layers.py", line 63, in
def _compute_fans(shape: jax.core.NamedShape, in_axis=-2, out_axis=-1):
File "c:\Users\mandar\AppData\Local\Programs\Python\Python310\lib\site-packages\jax_src\deprecations.py", line 57, in getattr
raise AttributeError(f"module {module!r} has no attribute {name!r}")
AttributeError: module 'jax.core' has no attribute 'NamedShape'