* 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>
be52722509