Commit 1a28741a authored by Gaetano Scandariato's avatar Gaetano Scandariato
Browse files

chore: Branch 2 reruns after the float32 fixes, cold-cache run, bias-vs-volume... 

chore: Branch 2 reruns after the float32 fixes, cold-cache run, bias-vs-volume diagnostic, juliet scatter from one source

Co-Authored-By: default avatarClaude Opus 5.5 <noreply@anthropic.com>
parent d71028a7

paper_benchmarks/gpu_precision/results/summary.json

+16 −0
Changes for paper_benchmarks/gpu_precision/results/summary.json: 16 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -258,5 +258,21 @@ 
      "command": "systemd-run --user --scope -p MemoryMax=8G -p MemorySwapMax=0 env JAX_ENABLE_COMPILATION_CACHE=false XLA_FLAGS=<flag> python scripts/stage.py llN pimen.yaml <batch>; host RSS from GNU time"

    },

    "issue": "https://github.com/openxla/xla/issues/49549"

  },

  "start_up_and_compilation_k2140_laptop_gpu": {

    "note": "K2-140 nested sampling on the laptop GPU, defaults of 2026-10-02; warm: compilation cache of earlier runs; cold: HOME pointed to an empty folder, so ~/.cache/pyaetna/jax_compilation was new (runs k2140_juliet/runs/k2c_e8_rtx and k2c_e8_rtx_cold); the nested-sampling loop time is from the run log",

    "warm": {

      "wall_s": 64.23,

      "nested_sampling_loop_s": 59.6

    },

    "cold": {

      "wall_s": 69.36,

      "nested_sampling_loop_s": 60.0

    },

    "outside_the_loop_s": {

      "warm": 4.630000000000003,

      "cold": 9.36

    },

    "identical_log_evidence": 14665.27

  }

}
 
 No newline at end of file

paper_benchmarks/k2140_juliet/results/summary.json

+9 −4
Changes for paper_benchmarks/k2140_juliet/results/summary.json: 9 added lines, 4 removed lines.
Original line number Diff line number Diff line
@@ -19,10 +19,15 @@ 
    "environment": "OMP_NUM_THREADS=1 OPENBLAS_NUM_THREADS=1, taskset -c 0-15 (2026-09-29)",

    "superseded_run": "reference/run_2026-09-25 (16 threads, OpenMP threads of batman not limited: 232 s, 1038587 calls)",

    "log_evidence_kms_runs": {

      "n_runs": 8,

      "mean": 14878.789608652001,

      "std": 2.244528740311011,

      "mean_reported_err": 0.51086475054653

      "n_runs": 14,

      "log_evidence_min": 14873.992505939263,

      "log_evidence_max": 14882.219063352897,

      "mean": 14879.143897574817,

      "mean_ms_units": 14665.00348392637,

      "scatter": 2.4642523539279044,

      "mean_reported_err": 0.5111583263176331,

      "scatter_over_reported": 4.820917956438845,

      "source": "cpu_scaling/results/summary.json"

    }

  },

  "pyaetna": {

paper_benchmarks/k2140_juliet/runs/k2c_e8_rtx_cold/PyAETNAouts/k2140_results.json

0 → 100644
+564 −0
Changes for paper_benchmarks/k2140_juliet/runs/k2c_e8_rtx_cold/PyAETNAouts/k2140_results.json: 564 added lines, 0 removed lines.
Original line number Diff line number Diff line 
{

 "format": "pyaetna-results",

 "format_version": 1,

 "evidence": {

  "log_evidence": 14665.26953125,

  "log_evidence_err": 0.3209257423877716,

  "information_nats": 48.2607421875,

  "converged": true,

  "n_iterations": 538,

  "n_dead": 27400,

  "n_likelihood_evals_approx": 5750711,

  "effective_sample_size": 4391.7900390625,

  "elapsed_s": 60.04836702346802,

  "prior_rejected_fraction": 0.0,

  "n_dim": 20,

  "num_inner_steps": 80,

  "num_delete": 50,

  "settings": {

   "n_live": 500,

   "dlogz": 0.1,

   "inner_steps_per_dim": 4,

   "delete_fraction": 0.1,

   "max_iterations": 100000,

   "num_samples": 4000,

   "space": "constrained",

   "slice_bracket": "adaptive",

   "slice_loop": "fused"

  },

  "posterior_summary": {

   "impact_b": {

    "median": 0.14618979394435883,

    "p16": 0.05146053433418274,

    "p84": 0.26670700311660767

   },

   "kRV_b": {

    "median": 103.71221923828125,

    "p16": 98.9794692993164,

    "p84": 108.37183380126953

   },

   "log_jitter_CORALIE": {

    "median": 0.13573488593101501,

    "p16": -1.5514789819717407,

    "p84": 1.932857632637024

   },

   "log_jitter_FIES": {

    "median": 0.5800303220748901,

    "p16": -1.306309461593628,

    "p84": 2.1245291233062744

   },

   "log_jitter_HARPS": {

    "median": 2.603416919708252,

    "p16": 1.1620484590530396,

    "p84": 3.1989479064941406

   },

   "log_jitter_K2": {

    "median": -7.869747638702393,

    "p16": -7.885392189025879,

    "p84": -7.854496955871582

   },

   "log_jitter_LCOGT": {

    "median": -10.809326171875,

    "p16": -12.840658187866211,

    "p84": -8.788935661315918

   },

   "norm_CORALIE": {

    "median": 1.2151122093200684,

    "p16": 1.20745849609375,

    "p84": 1.2225781679153442

   },

   "norm_FIES": {

    "median": 1.1318418979644775,

    "p16": 1.1280874013900757,

    "p84": 1.1354291439056396

   },

   "norm_HARPS": {

    "median": 1.2464466094970703,

    "p16": 1.2385324239730835,

    "p84": 1.2536976337432861

   },

   "norm_K2": {

    "median": 1.0000358819961548,

    "p16": 1.0000280141830444,

    "p84": 1.0000442266464233

   },

   "norm_LCOGT": {

    "median": 0.9998777508735657,

    "p16": 0.9996445178985596,

    "p84": 1.0001130104064941

   },

   "period_b": {

    "median": 6.569295406341553,

    "p16": 6.5692667961120605,

    "p84": 6.569324970245361

   },

   "q1_K2": {

    "median": 0.17073169350624084,

    "p16": 0.10905338078737259,

    "p84": 0.2982027232646942

   },

   "q1_LCOGT": {

    "median": 0.413690447807312,

    "p16": 0.23085370659828186,

    "p84": 0.6890127062797546

   },

   "q2_K2": {

    "median": 0.6155546307563782,

    "p16": 0.38422754406929016,

    "p84": 0.8574151396751404

   },

   "q2_LCOGT": {

    "median": 0.3636190891265869,

    "p16": 0.17183856666088104,

    "p84": 0.6519488096237183

   },

   "rho_star": {

    "median": 1.5572919845581055,

    "p16": 1.4549643993377686,

    "p84": 1.6219844818115234

   },

   "rp_by_rs_b": {

    "median": 0.1153256967663765,

    "p16": 0.11435084789991379,

    "p84": 0.1162649616599083

   },

   "tau_b": {

    "median": 0.0005891415057703853,

    "p16": 0.0003388885816093534,

    "p84": 0.0008381942170672119

   }

  },

  "device": "gpu",

  "precision": "float32",

  "random_seed": 1,

  "config_file": "k2140.yaml",

  "config_sha256": "152e9c695529756a409651c6a0c6a1f006bf6236f984743fcbf465f4cf41542b",

  "pyaetna_version": "3.8.0",

  "jax_version": "0.11.0",

  "timestamp_utc": "2026-10-03T06:51:12Z",

  "provenance": {

   "pyaetna_version": "3.8.0",

   "git_commit": "d71028a78474809572d71a2b67c1ba62d92a7c14",

   "git_dirty": false,

   "git_diff_sha256": null,

   "python": "3.14.4",

   "platform": "Linux-7.0.0-34-generic-x86_64-with-glibc2.43",

   "packages": {

    "jax": "0.11.0",

    "jaxlib": "0.11.0",

    "jax-cuda12-plugin": "0.11.0",

    "nvidia-cuda-runtime-cu12": "12.9.79",

    "numpyro": "0.22.0",

    "blackjax": "1.6.2",

    "tinygp": "0.3.1",

    "jaxoplanet": "0.1.0",

    "numpy": "2.5.1"

   },

   "jax_platforms": "cuda,cpu",

   "gpu": [

    {

     "name": "NVIDIA GeForce RTX 5070 Ti Laptop GPU",

     "driver": "595.91.07",

     "memory": "12227 MiB"

    }

   ],

   "compilation_cache_entries": null

  },

  "blackjax_version": "1.6.2"

 },

 "config_name": "k2140",

 "run": {

  "sampler": "ns",

  "device": "gpu",

  "precision": "float32",

  "num_chains": 1,

  "num_draws_per_chain": 4000,

  "num_warmup": 2000,

  "sampling_elapsed_s": 60.04836702346802,

  "random_seed": 1,

  "timestamp_utc": "2026-10-03T06:51:12Z"

 },

 "provenance": {

  "pyaetna_version": "3.8.0",

  "git_commit": "d71028a78474809572d71a2b67c1ba62d92a7c14",

  "git_dirty": false,

  "git_diff_sha256": null,

  "python": "3.14.4",

  "platform": "Linux-7.0.0-34-generic-x86_64-with-glibc2.43",

  "packages": {

   "jax": "0.11.0",

   "jaxlib": "0.11.0",

   "jax-cuda12-plugin": "0.11.0",

   "nvidia-cuda-runtime-cu12": "12.9.79",

   "numpyro": "0.22.0",

   "blackjax": "1.6.2",

   "tinygp": "0.3.1",

   "jaxoplanet": "0.1.0",

   "numpy": "2.5.1"

  },

  "jax_platforms": "cuda,cpu",

  "gpu": [

   {

    "name": "NVIDIA GeForce RTX 5070 Ti Laptop GPU",

    "driver": "595.91.07",

    "memory": "12227 MiB"

   }

  ],

  "compilation_cache_entries": null

 },

 "parameters": {

  "impact_b": {

   "mean": 0.15846003592014313,

   "std": 0.100925974547863,

   "ess": 3630.6000758463933,

   "rhat": 0.9997879266738892,

   "ess_per_sec": 60.46126240914241

  },

  "kRV_b": {

   "mean": 103.80783081054688,

   "std": 4.944850921630859,

   "ess": 3978.250125575116,

   "rhat": 0.9998818039894104,

   "ess_per_sec": 66.25076288952773

  },

  "log_jitter_CORALIE": {

   "mean": 0.2029150128364563,

   "std": 1.4992334842681885,

   "ess": 3961.8259233530625,

   "rhat": 1.0000046491622925,

   "ess_per_sec": 65.97724667191545

  },

  "log_jitter_FIES": {

   "mean": 0.4686252474784851,

   "std": 1.475030779838562,

   "ess": 3998.3620387705373,

   "rhat": 1.0002294778823853,

   "ess_per_sec": 66.58569145115808

  },

  "log_jitter_HARPS": {

   "mean": 2.211183547973633,

   "std": 1.3045393228530884,

   "ess": 4140.254335081589,

   "rhat": 0.9998874068260193,

   "ess_per_sec": 68.94865822851602

  },

  "log_jitter_K2": {

   "mean": -7.869852542877197,

   "std": 0.01589016616344452,

   "ess": 3885.2652312011865,

   "rhat": 0.9997533559799194,

   "ess_per_sec": 64.7022629221999

  },

  "log_jitter_LCOGT": {

   "mean": -10.804787635803223,

   "std": 1.761924386024475,

   "ess": 4207.400314054142,

   "rhat": 0.9999812841415405,

   "ess_per_sec": 70.06685647937456

  },

  "norm_CORALIE": {

   "mean": 1.2150588035583496,

   "std": 0.007781320251524448,

   "ess": 3935.2925952232317,

   "rhat": 1.0003113746643066,

   "ess_per_sec": 65.53538073208962

  },

  "norm_FIES": {

   "mean": 1.1317658424377441,

   "std": 0.00370796793140471,

   "ess": 3774.298667241394,

   "rhat": 1.000303864479065,

   "ess_per_sec": 62.85430985602536

  },

  "norm_HARPS": {

   "mean": 1.2459909915924072,

   "std": 0.00895415898412466,

   "ess": 3810.2704509494015,

   "rhat": 0.9998846650123596,

   "ess_per_sec": 63.453356682626804

  },

  "norm_K2": {

   "mean": 1.0000360012054443,

   "std": 8.198497198463883e-06,

   "ess": 3905.269635588253,

   "rhat": 1.0005956888198853,

   "ess_per_sec": 65.03540111360566

  },

  "norm_LCOGT": {

   "mean": 0.9998796582221985,

   "std": 0.00023355791927315295,

   "ess": 4099.609962202213,

   "rhat": 0.9998366832733154,

   "ess_per_sec": 68.27179764272373

  },

  "period_b": {

   "mean": 6.569295406341553,

   "std": 2.8726813980028965e-05,

   "ess": 4021.280861622848,

   "rhat": 1.0000253915786743,

   "ess_per_sec": 66.96736415914953

  },

  "q1_K2": {

   "mean": 0.20199458301067352,

   "std": 0.10928433388471603,

   "ess": 4035.259240987163,

   "rhat": 0.9997754096984863,

   "ess_per_sec": 67.20014949632368

  },

  "q1_LCOGT": {

   "mean": 0.4536339044570923,

   "std": 0.21379071474075317,

   "ess": 3742.8117143789955,

   "rhat": 0.9997506141662598,

   "ess_per_sec": 62.32995000374007

  },

  "q2_K2": {

   "mean": 0.6194062232971191,

   "std": 0.21097001433372498,

   "ess": 3784.758305017734,

   "rhat": 0.9997598528862,

   "ess_per_sec": 63.02849673728147

  },

  "q2_LCOGT": {

   "mean": 0.40078580379486084,

   "std": 0.23022542893886566,

   "ess": 3878.462020397698,

   "rhat": 0.9998132586479187,

   "ess_per_sec": 64.58896740492415

  },

  "rho_star": {

   "mean": 1.5379559993743896,

   "std": 0.0927424281835556,

   "ess": 3419.031328869305,

   "rhat": 0.9997717142105103,

   "ess_per_sec": 56.93795682292383

  },

  "rp_by_rs_b": {

   "mean": 0.11529670655727386,

   "std": 0.001016003079712391,

   "ess": 4101.923661532223,

   "rhat": 0.9997605085372925,

   "ess_per_sec": 68.31032823805376

  },

  "tau_b": {

   "mean": 0.0005863183760084212,

   "std": 0.00025494411238469183,

   "ess": 3806.256655531034,

   "rhat": 1.0022534132003784,

   "ess_per_sec": 63.386513975367194

  }

 },

 "fixed_params": {

  "norm_mode_K2": "multiplicative",

  "norm_mode_LCOGT": "multiplicative",

  "norm_mode_CORALIE": "additive",

  "norm_mode_FIES": "additive",

  "norm_mode_HARPS": "additive",

  "q1_CORALIE": 0.5,

  "q1_FIES": 0.5,

  "q1_HARPS": 0.5,

  "q2_CORALIE": 0.1,

  "q2_FIES": 0.1,

  "q2_HARPS": 0.1,

  "inc_star": 1.5707963267948966,

  "ecos_b": 0.0,

  "esin_b": 0.0,

  "amp_night_b": 0.0,

  "amp_day_minus_amp_night_b": 0.0,

  "amp_beam_b": 0.0,

  "amp_ellip_b": 0.0,

  "phoffset_b": 0.0,

  "tidallag_b": 0.0,

  "lambda_spin_b": 0.0

 },

 "map": null,

 "config": {

  "data": {

   "file_path": "k2140c.dat",

   "timeZeroPoint": 2457588.284,

   "working_dir": "./"

  },

  "instruments": [

   "K2",

   "LCOGT",

   "CORALIE",

   "FIES",

   "HARPS"

  ],

  "data_type": [

   "LC",

   "LC",

   "RV",

   "RV",

   "RV"

  ],

  "sampling": {

   "chain_energy_threshold": 5.0,

   "checkpoint_interval": 500,

   "internal_warmup_if_nuts": 500,

   "max_tree_depth": 10,

   "num_chains_if_nuts": 4,

   "num_samples_if_ensemble": 2000,

   "num_samples_if_nuts": 2000,

   "num_warmup": 2000,

   "production_sampler": "ns",

   "random_seed": 1,

   "reset": true,

   "target_accept_prob": 0.9,

   "use_float64": false,

   "walkers_per_free_parameter": 3,

   "warmup_sampler": "ess"

  },

  "detrending": {

   "normalization": [

    "per_instrument",

    [

     "normal",

     1.0,

     0.1

    ],

    [

     "normal",

     1.0,

     0.1

    ],

    [

     "normal",

     1.22,

     0.05

    ],

    [

     "normal",

     1.215,

     0.05

    ],

    [

     "normal",

     1.24,

     0.05

    ]

   ],

   "normalization_mode": [

    "per_instrument",

    "multiplicative",

    "multiplicative",

    "additive",

    "additive",

    "additive"

   ]

  },

  "log_jitter": [

   "per_instrument",

   [

    "uniform",

    -13.82,

    -7.6

   ],

   [

    "uniform",

    -13.82,

    -5.3

   ],

   [

    "uniform",

    -2.3,

    4.61

   ],

   [

    "uniform",

    -2.3,

    4.61

   ],

   [

    "uniform",

    -2.3,

    4.61

   ]

  ],

  "planet_b": {

   "amp_beam": 0.0,

   "amp_day_minus_amp_night": 0.0,

   "amp_ellip": 0.0,

   "amp_night": 0.0,

   "ecos": 0.0,

   "esin": 0.0,

   "impact": [

    "global",

    "uniform",

    0.0,

    1.2

   ],

   "kRV": [

    "global",

    "uniform",

    0.0,

    1000.0

   ],

   "period": [

    "global",

    "normal",

    6.5693,

    0.0001

   ],

   "phoffset": 0.0,

   "rp_by_rs": [

    "global",

    "uniform",

    0.01,

    0.5

   ],

   "tau": [

    "global",

    "normal",

    0.0,

    0.001

   ],

   "tidallag": 0.0

  },

  "reports": {

   "discard_extra_warmup": 0,

   "run": false,

   "trace_max_points": 500

  },

  "star": {

   "q1": [

    "per_instrument",

    [

     "uniform",

     0,

     1

    ],

    [

     "uniform",

     0,

     1

    ],

    0.5,

    0.5,

    0.5

   ],

   "q2": [

    "per_instrument",

    [

     "uniform",

     0,

     1

    ],

    [

     "uniform",

     0,

     1

    ],

    0.1,

    0.1,

    0.1

   ],

   "rho_star": [

    "global",

    "uniform",

    0.01,

    12.0

   ]

  }

 }

}
 
 No newline at end of file

paper_benchmarks/k2140_juliet/runs/k2c_e8_rtx_cold/PyAETNAouts/k2140_run.log

0 → 100644
+345 −0

File added.

Preview size limit exceeded, changes collapsed.

paper_benchmarks/k2140_juliet/runs/k2c_e8_rtx_cold/PyAETNAouts/k2140_samples.npz

0 → 100644
+234 KiB

File added.

No diff preview for this file type.

View file