From 25ec9dd009a03af8a4cfa653120b4a4abff506b9 Mon Sep 17 00:00:00 2001 From: requirements-bot Date: Tue, 19 May 2026 00:09:39 +0000 Subject: [PATCH] Committing all changes --- poetry.lock | 59 +++++++++++++++++++++++++++----------------- requirements-dev.txt | 2 +- requirements.txt | 2 +- 3 files changed, 39 insertions(+), 24 deletions(-) diff --git a/poetry.lock b/poetry.lock index 328592363..fda39a76b 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 2.2.1 and should not be changed by hand. +# This file is automatically @generated by Poetry 2.4.1 and should not be changed by hand. [[package]] name = "astropy" @@ -77,6 +77,7 @@ files = [ {file = "autograd-1.8.0-py3-none-any.whl", hash = "sha256:4ab9084294f814cf56c280adbe19612546a35574d67c574b04933c7d2ecb7d78"}, {file = "autograd-1.8.0.tar.gz", hash = "sha256:107374ded5b09fc8643ac925348c0369e7b0e73bbed9565ffd61b8fd04425683"}, ] +markers = {main = "extra == \"jax-support\""} [package.dependencies] numpy = "<3" @@ -415,7 +416,6 @@ description = "Python bindings for CUDA" optional = false python-versions = ">=3.10" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "cuda_bindings-13.2.0-cp310-cp310-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:08b395f79cb89ce0cd8effff07c4a1e20101b873c256a1aeb286e8fd7bd0f556"}, {file = "cuda_bindings-13.2.0-cp310-cp310-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:d6f3682ec3c4769326aafc67c2ba669d97d688d0b7e63e659d36d2f8b72f32d6"}, @@ -436,6 +436,7 @@ files = [ {file = "cuda_bindings-13.2.0-cp314-cp314t-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:e8a16384c6494e5485f39314b0b4afb04bee48d49edb16d5d8593fd35bbd231b"}, {file = "cuda_bindings-13.2.0-cp314-cp314t-win_amd64.whl", hash = "sha256:6ccf14e0c1def3b7200100aafff3a9f7e210ecb6e409329e92dcf6cd2c00d5c7"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [package.dependencies] cuda-pathfinder = ">=1.1,<2.0" @@ -450,10 +451,10 @@ description = "Pathfinder for CUDA components" optional = false python-versions = ">=3.10" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "cuda_pathfinder-1.3.3-py3-none-any.whl", hash = "sha256:9984b664e404f7c134954a771be8775dfd6180ea1e1aef4a5a37d4be05d9bbb1"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [[package]] name = "cuda-toolkit" @@ -462,10 +463,10 @@ description = "CUDA Toolkit meta-package" optional = false python-versions = "*" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "cuda_toolkit-13.0.2-py2.py3-none-any.whl", hash = "sha256:b198824cf2f54003f50d64ada3a0f184b42ca0846c1c94192fa269ecd97a66eb"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [package.dependencies] nvidia-cuda-cupti = {version = "==13.0.85.*", optional = true, markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"cupti\""} @@ -534,6 +535,7 @@ files = [ {file = "filelock-3.29.0-py3-none-any.whl", hash = "sha256:96f5f6344709aa1572bbf631c640e4ebeeb519e08da902c39a001882f30ac258"}, {file = "filelock-3.29.0.tar.gz", hash = "sha256:69974355e960702e789734cb4871f884ea6fe50bd8404051a3530bc07809cf90"}, ] +markers = {main = "extra == \"pytorch-support\""} [[package]] name = "fonttools" @@ -619,6 +621,7 @@ files = [ {file = "fsspec-2026.3.0-py3-none-any.whl", hash = "sha256:d2ceafaad1b3457968ed14efa28798162f1638dbb5d2a6868a2db002a5ee39a4"}, {file = "fsspec-2026.3.0.tar.gz", hash = "sha256:1ee6a0e28677557f8c2f994e3eea77db6392b4de9cd1f5d7a9e87a0ae9d01b41"}, ] +markers = {main = "extra == \"pytorch-support\""} [package.extras] abfs = ["adlfs"] @@ -726,6 +729,7 @@ files = [ {file = "healpy-1.19.0-cp314-cp314-musllinux_1_2_x86_64.whl", hash = "sha256:a103e327102e124785c1b5bdd379c6e5469137633212b56a6196bf04ddd100ff"}, {file = "healpy-1.19.0.tar.gz", hash = "sha256:28e839cb885a23d36c77fc3423a3cb9271a07fda94085bd12fc329f941130ec5"}, ] +markers = {main = "extra == \"healpy-support\""} [package.dependencies] astropy = "*" @@ -774,6 +778,7 @@ files = [ {file = "jax-0.10.0-py3-none-any.whl", hash = "sha256:76c42ba163c8db3dc2e449e225b888c0edfb623ded31efdc96d85e0fda1d26e8"}, {file = "jax-0.10.0.tar.gz", hash = "sha256:0119c767de1645f407df72345d28a3837dc904f1d698911c121d8f2b396fdece"}, ] +markers = {main = "extra == \"jax-support\""} [package.dependencies] jaxlib = "0.10.0" @@ -826,6 +831,7 @@ files = [ {file = "jaxlib-0.10.0-cp314-cp314t-manylinux_2_27_aarch64.whl", hash = "sha256:e66bdc0b57ed5649950799d3f0d67a6bb67f03d06b49ea3fced0bdd6140a9943"}, {file = "jaxlib-0.10.0-cp314-cp314t-manylinux_2_27_x86_64.whl", hash = "sha256:4dccd9065b30954879869641472d5d12fe4d7914175a5cad56293af8429ce7e0"}, ] +markers = {main = "extra == \"jax-support\""} [package.dependencies] ml_dtypes = ">=0.5.0" @@ -843,6 +849,7 @@ files = [ {file = "jinja2-3.1.6-py3-none-any.whl", hash = "sha256:85ece4451f492d0c13c5dd7c13a64681a86afae63a5f347908daf103ce6d2f67"}, {file = "jinja2-3.1.6.tar.gz", hash = "sha256:0137fb05990d35f1275a587e9aee6d56da821fc83491a0fb838183be43f66d6d"}, ] +markers = {main = "extra == \"pytorch-support\""} [package.dependencies] MarkupSafe = ">=2.0" @@ -1091,6 +1098,7 @@ files = [ {file = "markupsafe-3.0.3-cp39-cp39-win_arm64.whl", hash = "sha256:38664109c14ffc9e7437e86b4dceb442b0096dfe3541d7864d9cbe1da4cf36c8"}, {file = "markupsafe-3.0.3.tar.gz", hash = "sha256:722695808f4b6457b320fdc131280796bdceb04ab50fe1795cd540799ebe1698"}, ] +markers = {main = "extra == \"pytorch-support\""} [[package]] name = "matplotlib" @@ -1339,12 +1347,13 @@ files = [ {file = "ml_dtypes-0.5.4-cp39-cp39-win_amd64.whl", hash = "sha256:3d277bf3637f2a62176f4575512e9ff9ef51d00e39626d9fe4a161992f355af2"}, {file = "ml_dtypes-0.5.4.tar.gz", hash = "sha256:8ab06a50fb9bf9666dd0fe5dfb4676fa2b0ac0f31ecff72a6c3af8e22c063453"}, ] +markers = {main = "extra == \"jax-support\""} [package.dependencies] numpy = [ - {version = ">=2.1.0", markers = "python_version >= \"3.13\""}, - {version = ">=1.26.0", markers = "python_version == \"3.12\""}, {version = ">=1.23.3", markers = "python_version == \"3.11\""}, + {version = ">=1.26.0", markers = "python_version == \"3.12\""}, + {version = ">=2.1.0", markers = "python_version >= \"3.13\""}, ] [package.extras] @@ -1379,6 +1388,7 @@ files = [ {file = "networkx-3.6-py3-none-any.whl", hash = "sha256:cdb395b105806062473d3be36458d8f1459a4e4b98e236a66c3a48996e07684f"}, {file = "networkx-3.6.tar.gz", hash = "sha256:285276002ad1f7f7da0f7b42f004bcba70d381e936559166363707fdad3d72ad"}, ] +markers = {main = "extra == \"pytorch-support\""} [package.extras] benchmarking = ["asv", "virtualenv"] @@ -1480,12 +1490,12 @@ description = "CUBLAS native runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "nvidia_cublas-13.1.0.3-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:c86fc7f7ae36d7528288c5d88098edcb7b02c633d262e7ddbb86b0ad91be5df2"}, {file = "nvidia_cublas-13.1.0.3-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:ee8722c1f0145ab246bccb9e452153b5e0515fd094c3678df50b2a0888b8b171"}, {file = "nvidia_cublas-13.1.0.3-py3-none-win_amd64.whl", hash = "sha256:2a3b94a37def342471c59fad7856caee4926809a72dd5270155d6a31b5b277be"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [[package]] name = "nvidia-cuda-cupti" @@ -1494,12 +1504,12 @@ description = "CUDA profiling tools runtime libs." optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_cuda_cupti-13.0.85-py3-none-manylinux_2_25_aarch64.whl", hash = "sha256:796bd679890ee55fb14a94629b698b6db54bcfd833d391d5e94017dd9d7d3151"}, {file = "nvidia_cuda_cupti-13.0.85-py3-none-manylinux_2_25_x86_64.whl", hash = "sha256:4eb01c08e859bf924d222250d2e8f8b8ff6d3db4721288cf35d14252a4d933c8"}, {file = "nvidia_cuda_cupti-13.0.85-py3-none-win_amd64.whl", hash = "sha256:683f58d301548deeefcb8f6fac1b8d907691b9d8b18eccab417f51e362102f00"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [[package]] name = "nvidia-cuda-nvrtc" @@ -1508,12 +1518,12 @@ description = "NVRTC native runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_cuda_nvrtc-13.0.88-py3-none-manylinux2010_x86_64.manylinux_2_12_x86_64.whl", hash = "sha256:ad9b6d2ead2435f11cbb6868809d2adeeee302e9bb94bcf0539c7a40d80e8575"}, {file = "nvidia_cuda_nvrtc-13.0.88-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:d27f20a0ca67a4bb34268a5e951033496c5b74870b868bacd046b1b8e0c3267b"}, {file = "nvidia_cuda_nvrtc-13.0.88-py3-none-win_amd64.whl", hash = "sha256:6bcd4e7f8e205cbe644f5a98f2f799bef9556fefc89dd786e79a16312ce49872"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [[package]] name = "nvidia-cuda-runtime" @@ -1522,12 +1532,12 @@ description = "CUDA Runtime native Libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_cuda_runtime-13.0.96-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:ef9bcbe90493a2b9d810e43d249adb3d02e98dd30200d86607d8d02687c43f55"}, {file = "nvidia_cuda_runtime-13.0.96-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:7f82250d7782aa23b6cfe765ecc7db554bd3c2870c43f3d1821f1d18aebf0548"}, {file = "nvidia_cuda_runtime-13.0.96-py3-none-win_amd64.whl", hash = "sha256:f79298c8a098cec150a597c8eba58ecdab96e3bdc4b9bc4f9983635031740492"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [[package]] name = "nvidia-cudnn-cu13" @@ -1536,12 +1546,12 @@ description = "cuDNN runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "nvidia_cudnn_cu13-9.20.0.48-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:e31454ae00094b0c55319d9d15b6fa2fc50a9e1c0f5c8c80fb75258234e731e1"}, {file = "nvidia_cudnn_cu13-9.20.0.48-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:0c45dd8eeb50b603f07995b1b300c62ffe6a1980482b82b3bcf94a4ca9d49304"}, {file = "nvidia_cudnn_cu13-9.20.0.48-py3-none-win_amd64.whl", hash = "sha256:af8139732b99c0118be65ea5aac97f0d46018f8c552889e49d2fb0c6261a4a24"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [package.dependencies] nvidia-cublas = "*" @@ -1553,12 +1563,12 @@ description = "CUFFT native runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_cufft-12.0.0.61-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:2708c852ef8cd89d1d2068bdbece0aa188813a0c934db3779b9b1faa8442e5f5"}, {file = "nvidia_cufft-12.0.0.61-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:6c44f692dce8fd5ffd3e3df134b6cdb9c2f72d99cf40b62c32dde45eea9ddad3"}, {file = "nvidia_cufft-12.0.0.61-py3-none-win_amd64.whl", hash = "sha256:2abce5b39d2f5ae12730fb7e5db6696533e36c26e2d3e8fd1750bdd2853364eb"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [package.dependencies] nvidia-nvjitlink = "*" @@ -1570,11 +1580,11 @@ description = "cuFile GPUDirect libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "sys_platform == \"linux\" and platform_system == \"Linux\"" files = [ {file = "nvidia_cufile-1.15.1.6-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:08a3ecefae5a01c7f5117351c64f17c7c62efa5fffdbe24fc7d298da19cd0b44"}, {file = "nvidia_cufile-1.15.1.6-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:bdc0deedc61f548bddf7733bdc216456c2fdb101d020e1ab4b88d232d5e2f6d1"}, ] +markers = {main = "platform_system == \"Linux\" and sys_platform == \"linux\" and extra == \"pytorch-support\"", dev = "sys_platform == \"linux\" and platform_system == \"Linux\""} [[package]] name = "nvidia-curand" @@ -1583,12 +1593,12 @@ description = "CURAND native runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_curand-10.4.0.35-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:133df5a7509c3e292aaa2b477afd0194f06ce4ea24d714d616ff36439cee349a"}, {file = "nvidia_curand-10.4.0.35-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:1aee33a5da6e1db083fe2b90082def8915f30f3248d5896bcec36a579d941bfc"}, {file = "nvidia_curand-10.4.0.35-py3-none-win_amd64.whl", hash = "sha256:65b1710aa6961d326b411e314b374290904c5ddf41dc3f766ebc3f1d7d4ca69f"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [[package]] name = "nvidia-cusolver" @@ -1597,12 +1607,12 @@ description = "CUDA solver native runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_cusolver-12.0.4.66-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:02c2457eaa9e39de20f880f4bd8820e6a1cfb9f9a34f820eb12a155aa5bc92d2"}, {file = "nvidia_cusolver-12.0.4.66-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:0a759da5dea5c0ea10fd307de75cdeb59e7ea4fcb8add0924859b944babf1112"}, {file = "nvidia_cusolver-12.0.4.66-py3-none-win_amd64.whl", hash = "sha256:16515bd33a8e76bb54d024cfa068fa68d30e80fc34b9e1090813ea9362e0cb65"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [package.dependencies] nvidia-cublas = "*" @@ -1616,12 +1626,12 @@ description = "CUSPARSE native runtime libraries" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_cusparse-12.6.3.3-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:80bcc4662f23f1054ee334a15c72b8940402975e0eab63178fc7e670aa59472c"}, {file = "nvidia_cusparse-12.6.3.3-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:2b3c89c88d01ee0e477cb7f82ef60a11a4bcd57b6b87c33f789350b59759360b"}, {file = "nvidia_cusparse-12.6.3.3-py3-none-win_amd64.whl", hash = "sha256:cbcf42feb737bd7ec15b4c0a63e62351886bd3f975027b8815d7f720a2b5ea79"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [package.dependencies] nvidia-nvjitlink = "*" @@ -1633,12 +1643,12 @@ description = "NVIDIA cuSPARSELt" optional = false python-versions = "*" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "nvidia_cusparselt_cu13-0.8.1-py3-none-manylinux2014_aarch64.whl", hash = "sha256:4dca476c50bf4780d46cd0bfbd82e2bc10a08e4fef7950917ce8d7578d22a23f"}, {file = "nvidia_cusparselt_cu13-0.8.1-py3-none-manylinux2014_x86_64.whl", hash = "sha256:786ce87568c303fadb5afcc7102d454cd3040d75f6f8626f5db460d1871f4dd0"}, {file = "nvidia_cusparselt_cu13-0.8.1-py3-none-win_amd64.whl", hash = "sha256:dccbd362f91a7b9024d1f55ee9f548ac065027ff15d8c8b0db889ab3a8f31215"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [[package]] name = "nvidia-nccl-cu13" @@ -1647,11 +1657,11 @@ description = "NVIDIA Collective Communication Library (NCCL) Runtime" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "nvidia_nccl_cu13-2.29.7-py3-none-manylinux_2_18_aarch64.whl", hash = "sha256:674a12383e3c38a1bcccae7d4f3633b37852230b6047883cb2f4c2d1b36d9bf5"}, {file = "nvidia_nccl_cu13-2.29.7-py3-none-manylinux_2_18_x86_64.whl", hash = "sha256:edd81538446786ec3b73972543e53bb43bcaf0bfc8ef76cb679fcc390ffe136d"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [[package]] name = "nvidia-nvjitlink" @@ -1660,12 +1670,12 @@ description = "Nvidia JIT LTO Library" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_nvjitlink-13.0.88-py3-none-manylinux2010_x86_64.manylinux_2_12_x86_64.whl", hash = "sha256:13a74f429e23b921c1109976abefacc69835f2f433ebd323d3946e11d804e47b"}, {file = "nvidia_nvjitlink-13.0.88-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:e931536ccc7d467a98ba1d8b89ff7fa7f1fa3b13f2b0069118cd7f47bff07d0c"}, {file = "nvidia_nvjitlink-13.0.88-py3-none-win_amd64.whl", hash = "sha256:634e96e3da9ef845ae744097a1f289238ecf946ce0b82e93cdce14b9782e682f"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [[package]] name = "nvidia-nvshmem-cu13" @@ -1674,11 +1684,11 @@ description = "NVSHMEM creates a global address space that provides efficient an optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "nvidia_nvshmem_cu13-3.4.5-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:6dc2a197f38e5d0376ad52cd1a2a3617d3cdc150fd5966f4aee9bcebb1d68fe9"}, {file = "nvidia_nvshmem_cu13-3.4.5-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:290f0a2ee94c9f3687a02502f3b9299a9f9fe826e6d0287ee18482e78d495b80"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [[package]] name = "nvidia-nvtx" @@ -1687,12 +1697,12 @@ description = "NVIDIA Tools Extension" optional = false python-versions = ">=3" groups = ["main", "dev"] -markers = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\"" files = [ {file = "nvidia_nvtx-13.0.85-py3-none-manylinux1_x86_64.manylinux_2_5_x86_64.whl", hash = "sha256:4936d1d6780fbe68db454f5e72a42ff64d1fd6397df9f363ae786930fd5c1cd4"}, {file = "nvidia_nvtx-13.0.85-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:cb7780edb6b14107373c835bf8b72e7a178bac7367e23da7acb108f973f157a6"}, {file = "nvidia_nvtx-13.0.85-py3-none-win_amd64.whl", hash = "sha256:d66ea44254dd3c6eacc300047af6e1288d2269dd072b417e0adffbf479e18519"}, ] +markers = {main = "platform_system == \"Linux\" and (sys_platform == \"linux\" or sys_platform == \"win32\") and extra == \"pytorch-support\"", dev = "(sys_platform == \"linux\" or sys_platform == \"win32\") and platform_system == \"Linux\""} [[package]] name = "opt-einsum" @@ -1705,6 +1715,7 @@ files = [ {file = "opt_einsum-3.4.0-py3-none-any.whl", hash = "sha256:69bb92469f86a1565195ece4ac0323943e83477171b91d24c35afe028a90d7cd"}, {file = "opt_einsum-3.4.0.tar.gz", hash = "sha256:96ca72f1b886d148241348783498194c577fa30a8faac108586b14f1ba4473ac"}, ] +markers = {main = "extra == \"jax-support\""} [[package]] name = "packaging" @@ -2370,6 +2381,7 @@ files = [ {file = "setuptools-81.0.0-py3-none-any.whl", hash = "sha256:fdd925d5c5d9f62e4b74b30d6dd7828ce236fd6ed998a08d81de62ce5a6310d6"}, {file = "setuptools-81.0.0.tar.gz", hash = "sha256:487b53915f52501f0a79ccfd0c02c165ffe06631443a886740b91af4b7a5845a"}, ] +markers = {main = "extra == \"pytorch-support\""} [package.extras] check = ["pytest-checkdocs (>=2.4)", "pytest-ruff (>=0.2.1) ; sys_platform != \"cygwin\"", "ruff (>=0.13.0) ; sys_platform != \"cygwin\""] @@ -2477,6 +2489,7 @@ files = [ {file = "sympy-1.14.0-py3-none-any.whl", hash = "sha256:e091cc3e99d2141a0ba2847328f5479b05d94a6635cb96148ccb3f34671bd8f5"}, {file = "sympy-1.14.0.tar.gz", hash = "sha256:d3d3fe8df1e5a0b42f0e7bdf50541697dbe7d23746e894990c030e2b05e72517"}, ] +markers = {main = "extra == \"pytorch-support\""} [package.dependencies] mpmath = ">=1.1.0,<1.4" @@ -2521,6 +2534,7 @@ files = [ {file = "torch-2.12.0-cp314-cp314t-manylinux_2_28_x86_64.whl", hash = "sha256:6a7512adfdd7f6732e40de1c620831e3c75b39b98cef60b11d0c5f0a76473ec5"}, {file = "torch-2.12.0-cp314-cp314t-win_amd64.whl", hash = "sha256:5f96b63f8287f66a005dd1b5a6abba2920f11156c5e5c4d815f3e2050fd1aa16"}, ] +markers = {main = "extra == \"pytorch-support\""} [package.dependencies] cuda-bindings = {version = ">=13.0.3,<14", markers = "platform_system == \"Linux\""} @@ -2573,7 +2587,6 @@ description = "A language and compiler for custom Deep Learning operations" optional = false python-versions = "<3.15,>=3.10" groups = ["main", "dev"] -markers = "platform_system == \"Linux\"" files = [ {file = "triton-3.7.0-cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:223ac302091491436c248a34ee1e6c47a1026486579103c906ffd805be50cb89"}, {file = "triton-3.7.0-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:c631b65668d4951213b948a413c0564184305b77bb45cc9d686d3e1ecc4701a3"}, @@ -2590,6 +2603,7 @@ files = [ {file = "triton-3.7.0-cp314-cp314t-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:18a160de426fd99f92b0baf509045360afbd3bfaa0b4a5171dde800ec9f09684"}, {file = "triton-3.7.0-cp314-cp314t-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:ce061073102714b725f3660ec6939d94a1da7984b3aa99c921417cae273672f5"}, ] +markers = {main = "platform_system == \"Linux\" and extra == \"pytorch-support\"", dev = "platform_system == \"Linux\""} [package.extras] build = ["cmake (>=3.20,<4.0)", "lit"] @@ -2607,6 +2621,7 @@ files = [ {file = "typing_extensions-4.15.0-py3-none-any.whl", hash = "sha256:f0fa19c6845758ab08074a0cfa8b7aecb71c999ca73d62883bc25cc018c4e548"}, {file = "typing_extensions-4.15.0.tar.gz", hash = "sha256:0cea48d173cc12fa28ecabc3b837ea3cf6f38c6d1136f85cbaaf598984861466"}, ] +markers = {main = "extra == \"pytorch-support\""} [[package]] name = "tzdata" diff --git a/requirements-dev.txt b/requirements-dev.txt index b5d866d37..84e14b474 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -28,7 +28,7 @@ matplotlib==3.10.9 ; python_version >= "3.11" and python_version < "3.15" ml-dtypes==0.5.4 ; python_version >= "3.11" and python_version < "3.15" mpmath==1.3.0 ; python_version >= "3.11" and python_version < "3.15" networkx==3.6 ; python_version >= "3.11" and python_version < "3.15" -numpy==2.4.4 ; python_version >= "3.11" and python_version < "3.15" +numpy==2.4.5 ; python_version >= "3.11" and python_version < "3.15" nvidia-cublas==13.1.0.3 ; python_version >= "3.11" and python_version < "3.15" and platform_system == "Linux" nvidia-cuda-cupti==13.0.85 ; python_version >= "3.11" and python_version < "3.15" and platform_system == "Linux" and (sys_platform == "linux" or sys_platform == "win32") nvidia-cuda-nvrtc==13.0.88 ; python_version >= "3.11" and python_version < "3.15" and platform_system == "Linux" and (sys_platform == "linux" or sys_platform == "win32") diff --git a/requirements.txt b/requirements.txt index 9f4511340..2ecc3b288 100644 --- a/requirements.txt +++ b/requirements.txt @@ -22,7 +22,7 @@ markupsafe==3.0.3 ; python_version >= "3.11" and python_version < "3.15" matplotlib==3.10.9 ; python_version >= "3.11" and python_version < "3.15" mpmath==1.3.0 ; python_version >= "3.11" and python_version < "3.15" networkx==3.6 ; python_version >= "3.11" and python_version < "3.15" -numpy==2.4.4 ; python_version >= "3.11" and python_version < "3.15" +numpy==2.4.5 ; python_version >= "3.11" and python_version < "3.15" nvidia-cublas==13.1.0.3 ; python_version >= "3.11" and python_version < "3.15" and platform_system == "Linux" nvidia-cuda-cupti==13.0.85 ; python_version >= "3.11" and python_version < "3.15" and platform_system == "Linux" and (sys_platform == "linux" or sys_platform == "win32") nvidia-cuda-nvrtc==13.0.88 ; python_version >= "3.11" and python_version < "3.15" and platform_system == "Linux" and (sys_platform == "linux" or sys_platform == "win32")