Traceback (most recent call last):
File "D:/VS/python/nerfies-main/train.py", line 23, in
from flax import jax_utils
File "D:\RJ\A\python3.8\lib\site-packages\flax_init_.py", line 36, in
from . import core
File "D:\RJ\A\python3.8\lib\site-packages\flax\core_init_.py", line 15, in
from .axes_scan import broadcast
File "D:\RJ\A\python3.8\lib\site-packages\flax\core\axes_scan.py", line 17, in
import jax
File "D:\RJ\A\python3.8\lib\site-packages\jax_init_.py", line 109, in
from .experimental.maps import soft_pmap
File "D:\RJ\A\python3.8\lib\site-packages\jax\experimental\maps.py", line 25, in
from .. import numpy as jnp
File "D:\RJ\A\python3.8\lib\site-packages\jax\numpy_init_.py", line 16, in
from . import fft
File "D:\RJ\A\python3.8\lib\site-packages\jax\numpy\fft.py", line 17, in
from jax.src.numpy.fft import (
File "D:\RJ\A\python3.8\lib\site-packages\jax_src\numpy\fft.py", line 19, in
from jax import lax
File "D:\RJ\A\python3.8\lib\site-packages\jax\lax_init.py", line 332, in
from jax._src.lax.fft import (
File "D:\RJ\A\python3.8\lib\site-packages\jax_src\lax\fft.py", line 145, in
xla.backend_specific_translations['cpu'][fft_p] = pocketfft.pocketfft
AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft'