cuda_nvcc_path not found, TypeError in JAX 0.4.35

[fohmij]

I'm trying to get the GitHub repo turbozero by lowrollr running.
The pyproject.toml explicitly requires jax==0.4.35.

When I try to import jax, I get the following error:

TypeError: expected str, bytes or os.PathLike object, not NoneType

The error is raised at this line in jax/_src/lib/init.py:146:

cuda_nvcc_path = pathlib.Path(cuda_nvcc.__file__).parent

I'm working on a cluster with NVIDIA GPUs.
When I downgrade to jax==0.4.34, the error disappears — because that version doesn't try to access the nvcc path.

Any advice on how to fix or bypass this issue while staying on JAX 0.4.35 would be appreciated.