Files
HonorThesis/codes/experiment_v0.2/requirements.txt
2026-03-11 12:31:12 -04:00

11 lines
210 B
Plaintext

numpy>=1.26
matplotlib>=3.8
tqdm>=4.66
# CPU-only JAX
# jax
# Apple Metal JAX (experimental; complex64/complex128 currently unsupported)
# jax-metal
# NVIDIA Linux JAX
jax[cuda13]
# or, if needed:
# jax[cuda12]