24 lines
973 B
Python
24 lines
973 B
Python
"""Edit globals here; no CLI parser is used."""
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
SEED = 7
|
|
KAPPA = 1e-3
|
|
NUM_SAMPLES = 10**4 # requested default
|
|
LIPSCHITZ_PAIRS = 12_000
|
|
LIPSCHITZ_RESERVOIR = 4_096
|
|
MAJORANA_STAR_STATES = 16 # only for visualization
|
|
MAX_STAR_DEGREE = 63 # avoid unstable huge root-finding plots
|
|
|
|
BACKEND = "auto" # auto | jax | numpy
|
|
JAX_PLATFORM = "" # "", "cpu", "gpu"; set before importing JAX
|
|
RESULTS_DIR = Path("./results") / f"exp-{datetime.now():%Y%m%d-%H%M%S}"
|
|
|
|
# Chosen so the three families have comparable intrinsic dimensions:
|
|
# sphere S^(m-1), CP^(d_A d_B - 1), and Sym^N(C^2) ~ CP^N.
|
|
SPHERE_DIMS = [16, 64, 256, 1024]
|
|
CP_DIMS = [(4, 4), (8, 8), (16, 16), (32, 32)]
|
|
MAJORANA_N = [15, 63, 255, 1023]
|
|
|
|
# Batch sizes are the main speed knob; reduce CP batches first if memory is tight.
|
|
BATCH = {"sphere": 32_768, "cp": 256, "majorana": 65_536} |