There are many different versions of the `cudatoolkit` and related
cuda packages, and it can be tricky to ensure they remain compatible.
- `cudaPackages` is now a package set with `cudatoolkit`, `cudnn`, `cutensor`, `nccl`, as well as `cudatoolkit` split into smaller packages ("redist");
- expressions should now use `cudaPackages` as parameter instead of the individual cuda packages;
- `makeScope` is now used, so it is possible to use `.overrideScope'` to set e.g. a different `cudnn` version;
- `release-cuda.nix` is introduced to easily evaluate cuda packages using hydra.
* python3Packages.jaxlib: rename to `jaxlib-bin`
Refactoring `jaxlib` to have a similar structure to `tensorflow` with the 'bin' and 'build' options.
* python3Packages.jaxlib: init the 'build' variant at 0.1.75
Similar to `tensorflow-build`, now there's an option to build `jaxlib` using Nix-provided environment and dependencies.
* python3Packages.jax: 0.2.24 -> 0.2.26
* Addressed review comments.
* Fixed `cudaSupport` missing property on some arches.
* Unified the versions of CUDA-related packages with TF.
Co-authored-by: Samuel Ainsworth <skainsworth@gmail.com>
