diff --git a/jax_profiling/results/jit/datacube/delaunay/alma_high/comparison.json b/jax_profiling/results/jit/datacube/delaunay/alma_high/comparison.json new file mode 100644 index 0000000..9081f16 --- /dev/null +++ b/jax_profiling/results/jit/datacube/delaunay/alma_high/comparison.json @@ -0,0 +1,136 @@ +{ + "configs": { + "hpc_a100_fp64": { + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33287 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "n_channels": 34, + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities_per_channel": 5000000, + "hilbert_pixels": 500, + "delaunay_vertices": 500, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "cube_log_evidence_eager": -2048222823.6831546, + "cube_log_evidence_jit": null, + "log_evidence_per_channel_eager": [ + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695 + ], + "full_pipeline_cube_single_jit": NaN, + "shared_lwl_savings_estimate": null, + "vmap": "SKIPPED \u2014 cube batching axis is 'datasets', not 'parameters'", + "config_name": "hpc_a100_fp64" + }, + "hpc_a100_mp": { + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33287 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "n_channels": 34, + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities_per_channel": 5000000, + "hilbert_pixels": 500, + "delaunay_vertices": 500, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "cube_log_evidence_eager": -2048222823.6695518, + "cube_log_evidence_jit": null, + "log_evidence_per_channel_eager": [ + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686 + ], + "full_pipeline_cube_single_jit": NaN, + "shared_lwl_savings_estimate": null, + "vmap": "SKIPPED \u2014 cube batching axis is 'datasets', not 'parameters'", + "config_name": "hpc_a100_mp" + } + } +} \ No newline at end of file diff --git a/jax_profiling/results/jit/datacube/delaunay/alma_high/hpc_a100_fp64.json b/jax_profiling/results/jit/datacube/delaunay/alma_high/hpc_a100_fp64.json new file mode 100644 index 0000000..c6b32fa --- /dev/null +++ b/jax_profiling/results/jit/datacube/delaunay/alma_high/hpc_a100_fp64.json @@ -0,0 +1,65 @@ +{ + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33287 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "n_channels": 34, + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities_per_channel": 5000000, + "hilbert_pixels": 500, + "delaunay_vertices": 500, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "cube_log_evidence_eager": -2048222823.6831546, + "cube_log_evidence_jit": null, + "log_evidence_per_channel_eager": [ + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695, + -60241847.75538695 + ], + "full_pipeline_cube_single_jit": NaN, + "shared_lwl_savings_estimate": null, + "vmap": "SKIPPED \u2014 cube batching axis is 'datasets', not 'parameters'" +} \ No newline at end of file diff --git a/jax_profiling/results/jit/datacube/delaunay/alma_high/hpc_a100_mp.json b/jax_profiling/results/jit/datacube/delaunay/alma_high/hpc_a100_mp.json new file mode 100644 index 0000000..010799a --- /dev/null +++ b/jax_profiling/results/jit/datacube/delaunay/alma_high/hpc_a100_mp.json @@ -0,0 +1,65 @@ +{ + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33287 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "n_channels": 34, + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities_per_channel": 5000000, + "hilbert_pixels": 500, + "delaunay_vertices": 500, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "cube_log_evidence_eager": -2048222823.6695518, + "cube_log_evidence_jit": null, + "log_evidence_per_channel_eager": [ + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686, + -60241847.75498686 + ], + "full_pipeline_cube_single_jit": NaN, + "shared_lwl_savings_estimate": null, + "vmap": "SKIPPED \u2014 cube batching axis is 'datasets', not 'parameters'" +} \ No newline at end of file diff --git a/jax_profiling/results/jit/interferometer/delaunay/alma_high/comparison.json b/jax_profiling/results/jit/interferometer/delaunay/alma_high/comparison.json new file mode 100644 index 0000000..43e208c --- /dev/null +++ b/jax_profiling/results/jit/interferometer/delaunay/alma_high/comparison.json @@ -0,0 +1,66 @@ +{ + "configs": { + "hpc_a100_fp64": { + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33441 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities": 5000000, + "hilbert_pixels": 1000, + "delaunay_vertices": 1000, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "log_likelihood_eager": -60238977.636567496, + "figure_of_merit_eager": -60243535.85638879, + "log_evidence_jit": -60243535.85638878, + "full_pipeline_single_jit": 0.09768839629832655, + "vmap": "SKIPPED \u2014 opt-in via DELAUNAY_VMAP=1", + "memory_mb": null, + "full_pipeline_per_call": 0.09768839629832655, + "config_name": "hpc_a100_fp64" + }, + "hpc_a100_mp": { + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33441 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities": 5000000, + "hilbert_pixels": 1000, + "delaunay_vertices": 1000, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "log_likelihood_eager": -60238977.636564635, + "figure_of_merit_eager": -60243535.85614109, + "log_evidence_jit": -60243535.85614108, + "full_pipeline_single_jit": 0.10049871581140905, + "vmap": "SKIPPED \u2014 opt-in via DELAUNAY_VMAP=1", + "memory_mb": null, + "full_pipeline_per_call": 0.10049871581140905, + "config_name": "hpc_a100_mp" + } + } +} \ No newline at end of file diff --git a/jax_profiling/results/jit/interferometer/delaunay/alma_high/comparison.png b/jax_profiling/results/jit/interferometer/delaunay/alma_high/comparison.png new file mode 100644 index 0000000..6a9b4ee Binary files /dev/null and b/jax_profiling/results/jit/interferometer/delaunay/alma_high/comparison.png differ diff --git a/jax_profiling/results/jit/interferometer/delaunay/alma_high/hpc_a100_fp64.json b/jax_profiling/results/jit/interferometer/delaunay/alma_high/hpc_a100_fp64.json new file mode 100644 index 0000000..11040fe --- /dev/null +++ b/jax_profiling/results/jit/interferometer/delaunay/alma_high/hpc_a100_fp64.json @@ -0,0 +1,29 @@ +{ + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33441 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities": 5000000, + "hilbert_pixels": 1000, + "delaunay_vertices": 1000, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "log_likelihood_eager": -60238977.636567496, + "figure_of_merit_eager": -60243535.85638879, + "log_evidence_jit": -60243535.85638878, + "full_pipeline_single_jit": 0.09768839629832655, + "vmap": "SKIPPED \u2014 opt-in via DELAUNAY_VMAP=1", + "memory_mb": null +} \ No newline at end of file diff --git a/jax_profiling/results/jit/interferometer/delaunay/alma_high/hpc_a100_mp.json b/jax_profiling/results/jit/interferometer/delaunay/alma_high/hpc_a100_mp.json new file mode 100644 index 0000000..c984ce5 --- /dev/null +++ b/jax_profiling/results/jit/interferometer/delaunay/alma_high/hpc_a100_mp.json @@ -0,0 +1,29 @@ +{ + "autolens_version": "2026.5.14.2", + "device": { + "backend": "gpu", + "device": "cuda:0", + "nvidia_smi": "NVIDIA A100 80GB PCIe, 33441 MiB, 81920 MiB" + }, + "instrument": "alma_high", + "model": "delaunay", + "configuration": { + "pixel_scale_arcsec": 0.025, + "mask_radius_arcsec": 3.5, + "real_space_shape": [ + 800, + 800 + ], + "visibilities": 5000000, + "hilbert_pixels": 1000, + "delaunay_vertices": 1000, + "edge_zeroed_pixels": 0, + "regularization_coefficient": 1.0 + }, + "log_likelihood_eager": -60238977.636564635, + "figure_of_merit_eager": -60243535.85614109, + "log_evidence_jit": -60243535.85614108, + "full_pipeline_single_jit": 0.10049871581140905, + "vmap": "SKIPPED \u2014 opt-in via DELAUNAY_VMAP=1", + "memory_mb": null +} \ No newline at end of file