ERROR: No matching distribution found for jax-cuda12-pjrt

[asifiqbal]

Description of the bug:

on macos Apple M2 Pro

❯ bash setup_python.sh
Collecting absl-py (from -r requirements.txt (line 1))
  Using cached absl_py-2.3.1-py3-none-any.whl.metadata (3.3 kB)
Collecting annotated-types (from -r requirements.txt (line 2))
  Using cached annotated_types-0.7.0-py3-none-any.whl.metadata (15 kB)
Collecting anyio (from -r requirements.txt (line 3))
  Using cached anyio-4.10.0-py3-none-any.whl.metadata (4.0 kB)
Collecting astunparse (from -r requirements.txt (line 4))
  Using cached astunparse-1.6.3-py2.py3-none-any.whl.metadata (4.4 kB)
Collecting certifi (from -r requirements.txt (line 5))
  Using cached certifi-2025.8.3-py3-none-any.whl.metadata (2.4 kB)
Collecting charset-normalizer (from -r requirements.txt (line 6))
  Using cached charset_normalizer-3.4.3-cp313-cp313-macosx_10_13_universal2.whl.metadata (36 kB)
Collecting click (from -r requirements.txt (line 7))
  Using cached click-8.2.1-py3-none-any.whl.metadata (2.5 kB)
Collecting exceptiongroup (from -r requirements.txt (line 8))
  Using cached exceptiongroup-1.3.0-py3-none-any.whl.metadata (6.7 kB)
Collecting fastapi (from -r requirements.txt (line 9))
  Using cached fastapi-0.116.1-py3-none-any.whl.metadata (28 kB)
Collecting flatbuffers (from -r requirements.txt (line 10))
  Using cached flatbuffers-25.2.10-py2.py3-none-any.whl.metadata (875 bytes)
Collecting gast (from -r requirements.txt (line 11))
  Using cached gast-0.6.0-py3-none-any.whl.metadata (1.3 kB)
Collecting google-pasta (from -r requirements.txt (line 12))
  Using cached google_pasta-0.2.0-py3-none-any.whl.metadata (814 bytes)
Collecting grpcio (from -r requirements.txt (line 13))
  Using cached grpcio-1.74.0-cp313-cp313-macosx_11_0_universal2.whl.metadata (3.8 kB)
Collecting h11 (from -r requirements.txt (line 14))
  Using cached h11-0.16.0-py3-none-any.whl.metadata (8.3 kB)
Collecting h5py (from -r requirements.txt (line 15))
  Using cached h5py-3.14.0-cp313-cp313-macosx_11_0_arm64.whl.metadata (2.7 kB)
Collecting httptools (from -r requirements.txt (line 16))
  Using cached httptools-0.6.4-cp313-cp313-macosx_11_0_arm64.whl.metadata (3.6 kB)
Collecting idna (from -r requirements.txt (line 17))
  Using cached idna-3.10-py3-none-any.whl.metadata (10 kB)
Collecting jax (from -r requirements.txt (line 18))
  Using cached jax-0.7.1-py3-none-any.whl.metadata (13 kB)
ERROR: Could not find a version that satisfies the requirement jax-cuda12-pjrt (from versions: none)
ERROR: No matching distribution found for jax-cuda12-pjrt

Actual vs expected behavior:

No response

Any other information you'd like to share?

No response

[shawnlanum368-crypto]

Disable errors an blocks