Building from soure fails with CUDA

[GaetanLepage]

Description

Context: jax update to 0.4.25 in the nixpkgs repository

Building the latest version (0.4.25) works fine without CUDA support. However, when enabling the latter, we face:

ERROR: /build/source/jaxlib/cuda/BUILD:185:17: no such package '@tsl//tsl/cuda': BUILD file not found in directory 'tsl/cuda>
ERROR: Analysis of target '//jaxlib/tools:build_wheel' failed; build aborted: 

Can this be our fault ?

cc @samuela @SomeoneSerge

System info (python version, jaxlib version, accelerator, etc.)

Python version: 3.11 CUDA version: 11.8 (also tried with 12.2)