···47804781 jaxlib-bin = callPackage ../development/python-modules/jaxlib/bin.nix {
4782 cudaSupport = pkgs.config.cudaSupport or false;
4783- # At the time of writing (2022-04-18), `cudaPackages.nccl` is broken, so we
4784- # pin to `cudaPackages_11_6` instead.
4785- cudaPackages = pkgs.cudaPackages_11_6;
4786 };
47874788 jaxlib-build = callPackage ../development/python-modules/jaxlib rec {
···4792 };
4793 # Some platforms don't have `cudaSupport` defined, hence the need for 'or false'.
4794 cudaSupport = pkgs.config.cudaSupport or false;
4795- # At the time of writing (2022-04-18), `cudaPackages.nccl` is broken, so we
4796- # pin to `cudaPackages_11_6` instead.
4797- cudaPackages = pkgs.cudaPackages_11_6;
4798 IOKit = pkgs.darwin.apple_sdk_11_0.IOKit;
4799 protobuf = pkgs.protobuf3_20; # jaxlib-build 0.3.15 won't build with protobuf 3.21
4800 };
···47804781 jaxlib-bin = callPackage ../development/python-modules/jaxlib/bin.nix {
4782 cudaSupport = pkgs.config.cudaSupport or false;
0004783 };
47844785 jaxlib-build = callPackage ../development/python-modules/jaxlib rec {
···4789 };
4790 # Some platforms don't have `cudaSupport` defined, hence the need for 'or false'.
4791 cudaSupport = pkgs.config.cudaSupport or false;
0004792 IOKit = pkgs.darwin.apple_sdk_11_0.IOKit;
4793 protobuf = pkgs.protobuf3_20; # jaxlib-build 0.3.15 won't build with protobuf 3.21
4794 };