@@ -31,8 +31,8 @@ classifiers = [
3131]
3232requires-python = " >=3.10"
3333dependencies = [
34- " numpy >= 1.21.0" ,
35- " scipy >= 1.11.0" ,
34+ " numpy >= 1.21.0; extra != 'intel' " ,
35+ " scipy >= 1.11.0; extra != 'intel' " ,
3636]
3737dynamic = [" version" ]
3838
@@ -55,6 +55,11 @@ deep = ["torch", "jax"]
5555deep-cu126 = [" torch<2.11.0" , " jax[cuda12]" ]
5656deep-cu128 = [" torch<2.11.0" , " jax[cuda12]" ]
5757deep-cu13 = [" torch>=2.11.0" , " jax[cuda13]" ]
58+ intel = [
59+ " numpy>=2.0.0; sys_platform != 'darwin'" ,
60+ " scipy>=1.13.0; sys_platform != 'darwin'" ,
61+ " mkl_fft; sys_platform != 'darwin'" ,
62+ ]
5863
5964[dependency-groups ]
6065dev = [
@@ -111,6 +116,9 @@ torch = [
111116 { index = " pytorch-cu126" , extra = " deep-cu126" },
112117 { index = " pytorch-cu128" , extra = " deep-cu128" },
113118]
119+ numpy = [{ index = " intel" , extra = " intel" }]
120+ scipy = [{ index = " intel" , extra = " intel" }]
121+ mkl_fft = [{ index = " intel" , extra = " intel" }]
114122
115123[[tool .uv .index ]]
116124name = " pytorch-cpu"
@@ -127,6 +135,11 @@ name = "pytorch-cu128"
127135url = " https://download.pytorch.org/whl/cu128"
128136explicit = true
129137
138+ [[tool .uv .index ]]
139+ name = " intel"
140+ url = " https://software.repos.intel.com/python/pypi"
141+ explicit = true
142+
130143[tool .pytest .ini_options ]
131144addopts = " --verbose"
132145python_files = [" pytests/*.py" ]
0 commit comments