diff --git a/pixi.lock b/pixi.lock index ab1106b1..1d055b44 100644 --- a/pixi.lock +++ b/pixi.lock @@ -1416,7 +1416,7 @@ environments: - conda: https://prefix.dev/conda-forge/linux-64/gmp-6.3.0-hac33072_2.conda - conda: https://prefix.dev/conda-forge/linux-64/gmpy2-2.3.0-py314h28848ee_1.conda - conda: https://prefix.dev/conda-forge/linux-64/icu-78.3-h33c6efd_0.conda - - conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.1-cuda129_py314hfcd48a0_200.conda + - conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.2-cuda129_py314hfcd48a0_200.conda - conda: https://prefix.dev/conda-forge/linux-64/ld_impl_linux-64-2.45.1-default_hbd61a6d_102.conda - conda: https://prefix.dev/conda-forge/linux-64/lefthook-2.1.9-hfc2019e_0.conda - conda: https://prefix.dev/conda-forge/linux-64/libabseil-20260107.1-cxx17_h7b12aa8_0.conda @@ -4849,7 +4849,7 @@ environments: - conda: https://prefix.dev/conda-forge/linux-64/gmp-6.3.0-hac33072_2.conda - conda: https://prefix.dev/conda-forge/linux-64/gmpy2-2.3.0-py314h28848ee_1.conda - conda: https://prefix.dev/conda-forge/linux-64/icu-78.3-h33c6efd_0.conda - - conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.1-cuda129_py314hfcd48a0_200.conda + - conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.2-cuda129_py314hfcd48a0_200.conda - conda: https://prefix.dev/conda-forge/linux-64/ld_impl_linux-64-2.45.1-default_hbd61a6d_102.conda - conda: https://prefix.dev/conda-forge/linux-64/libabseil-20260107.1-cxx17_h7b12aa8_0.conda - conda: https://prefix.dev/conda-forge/linux-64/libblas-3.11.0-8_h5875eb1_mkl.conda @@ -5115,7 +5115,7 @@ environments: - conda: https://prefix.dev/conda-forge/linux-64/gmp-6.3.0-hac33072_2.conda - conda: https://prefix.dev/conda-forge/linux-64/gmpy2-2.3.0-py311h92a432a_1.conda - conda: https://prefix.dev/conda-forge/linux-64/icu-78.3-h33c6efd_0.conda - - conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.1-cuda129_py311h6689f8c_200.conda + - conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.2-cuda129_py311h6689f8c_200.conda - conda: https://prefix.dev/conda-forge/linux-64/ld_impl_linux-64-2.45.1-default_hbd61a6d_102.conda - conda: https://prefix.dev/conda-forge/linux-64/libabseil-20260107.1-cxx17_h7b12aa8_0.conda - conda: https://prefix.dev/conda-forge/linux-64/libblas-3.11.0-8_h5875eb1_mkl.conda @@ -5246,7 +5246,7 @@ environments: - conda: https://prefix.dev/conda-forge/noarch/typing_extensions-4.15.0-pyhcf101f3_0.conda - conda: https://prefix.dev/conda-forge/noarch/tzdata-2025c-hc9c84f9_1.conda - conda: https://prefix.dev/conda-forge/noarch/zipp-4.1.0-pyhcf101f3_0.conda - - conda_source: array-api-extra[0d36b293] @ . + - conda_source: array-api-extra[08d18e4a] @ . p2: - conda: https://prefix.dev/conda-forge/noarch/array-api-compat-1.15.0-pyhc364b38_0.conda - conda: https://prefix.dev/conda-forge/noarch/array-api-strict-2.5-pyhcf101f3_0.conda @@ -7688,6 +7688,104 @@ packages: run_exports: {} size: 208900015 timestamp: 1781099249302 +- conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.2-cuda129_py311h6689f8c_200.conda + sha256: a6e2ba1fdc24c822dd96df56f2112da54238bddd1a58d0cadd03dce877b00e3d + md5: 854dddcef3c0558e7956bfef4a6941c9 + depends: + - python + - scipy >=1.9 + - ml_dtypes >=0.2.0 + - onednn-cpu-threadpool + - __cuda + - cuda-nvcc-tools + - cuda-cupti-dev + - libcublas-dev + - libcufft-dev + - libcurand-dev + - libcusolver-dev + - libcusparse-dev + - cuda-version >=12.9,<13 + - libgcc >=15 + - libstdcxx >=15 + - __glibc >=2.17,<3.0.a0 + - flatbuffers >=25.9.23,<25.9.24.0a0 + - cuda-cupti >=12.9.79,<13.0a0 + - libcurand >=10.3.10.19,<11.0a0 + - libabseil >=20260107.1,<20260108.0a0 + - libabseil * cxx17* + - libcusolver >=11.7.5.82,<12.0a0 + - cudnn >=9.10.2.21,<10.0a0 + - numpy >=1.23,<3 + - onednn >=3.12,<4.0a0 + - libgrpc >=1.78.1,<1.79.0a0 + - cuda-nvtx >=12.9.79,<13.0a0 + - python_abi 3.11.* *_cp311 + - libcublas >=12.9.2.10,<13.0a0 + - libcufft >=11.4.1.4,<12.0a0 + - libre2-11 >=2025.11.5 + - re2 + - cuda-cudart >=12.9.79,<13.0a0 + - cuda-nvrtc >=12.9.86,<13.0a0 + - libcusparse >=12.5.10.65,<13.0a0 + - nccl >=2.30.7.1,<3.0a0 + - openssl >=3.5.7,<4.0a0 + - libzlib >=1.3.2,<2.0a0 + constrains: + - jax >=0.10.2 + license: Apache-2.0 + license_family: APACHE + run_exports: {} + size: 209195433 + timestamp: 1783084339134 +- conda: https://prefix.dev/conda-forge/linux-64/jaxlib-0.10.2-cuda129_py314hfcd48a0_200.conda + sha256: 0a2c4cd4762652e0206697fdcab27553b7af15e7e6e75a80bd42d15f447a1f5a + md5: c050bfd9ebd31e143ac24335c33354be + depends: + - python + - scipy >=1.9 + - ml_dtypes >=0.2.0 + - onednn-cpu-threadpool + - __cuda + - cuda-nvcc-tools + - cuda-cupti-dev + - libcublas-dev + - libcufft-dev + - libcurand-dev + - libcusolver-dev + - libcusparse-dev + - __glibc >=2.17,<3.0.a0 + - libstdcxx >=15 + - libgcc >=15 + - cuda-version >=12.9,<13 + - libcurand >=10.3.10.19,<11.0a0 + - libabseil >=20260107.1,<20260108.0a0 + - libabseil * cxx17* + - flatbuffers >=25.9.23,<25.9.24.0a0 + - cudnn >=9.10.2.21,<10.0a0 + - nccl >=2.30.7.1,<3.0a0 + - libcublas >=12.9.2.10,<13.0a0 + - libre2-11 >=2025.11.5 + - re2 + - libgrpc >=1.78.1,<1.79.0a0 + - openssl >=3.5.7,<4.0a0 + - cuda-cupti >=12.9.79,<13.0a0 + - cuda-nvrtc >=12.9.86,<13.0a0 + - libcusparse >=12.5.10.65,<13.0a0 + - onednn >=3.12,<4.0a0 + - python_abi 3.14.* *_cp314 + - cuda-nvtx >=12.9.79,<13.0a0 + - numpy >=1.25,<3 + - cuda-cudart >=12.9.79,<13.0a0 + - libzlib >=1.3.2,<2.0a0 + - libcufft >=11.4.1.4,<12.0a0 + - libcusolver >=11.7.5.82,<12.0a0 + constrains: + - jax >=0.10.2 + license: Apache-2.0 + license_family: APACHE + run_exports: {} + size: 209314320 + timestamp: 1783084338485 - conda: https://prefix.dev/conda-forge/linux-64/ld_impl_linux-64-2.45.1-default_hbd61a6d_102.conda sha256: 3d584956604909ff5df353767f3a2a2f60e07d070b328d109f30ac40cd62df6c md5: 18335a698559cdbcd86150a48bf54ba6 diff --git a/pixi.toml b/pixi.toml index c70a0682..24f017ab 100644 --- a/pixi.toml +++ b/pixi.toml @@ -197,7 +197,7 @@ cupy = ">=14.1.1" pytorch = { version = ">=2.12.0", build = "cuda12*" } [feature.cuda-backends.target.linux.dependencies] -jaxlib = { version = ">=0.10.1", build = "cuda12*" } +jaxlib = { version = ">=0.10.2", build = "cuda12*" } [feature.nogil.dependencies] python-freethreading = "~=3.13.0"