-
Notifications
You must be signed in to change notification settings - Fork 12
Expand file tree
/
Copy pathrequirements.txt
More file actions
29 lines (24 loc) · 1.1 KB
/
Copy pathrequirements.txt
File metadata and controls
29 lines (24 loc) · 1.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# ─────────────────────────────────────────────────────────────────────────────
# CPU-only install (Windows native, macOS):
# pip install -r requirements.txt
#
# GPU/CUDA install (WSL2 + NVIDIA GPU — RECOMMENDED):
# pip install "jax[cuda12]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
# pip install matplotlib pytest
#
# TPU install (Google Cloud TPU v5e-16):
# pip install -r requirements_tpu.txt
# (see requirements_tpu.txt for jax[tpu] with libtpu)
# ─────────────────────────────────────────────────────────────────────────────
# Core
jax>=0.10.0
jaxlib>=0.10.0
# Numerics & Scientific
numpy>=2.0.0
scipy>=1.14.0
# Visualization
matplotlib>=3.10.0
# Number theory (optional — used by Shor's algorithm; stdlib gcd/fractions also works)
# sympy>=1.12
# Testing
pytest>=9.0.0