{ "metadata": { "cpu_summary_paths": [ "outputs/fixed_runtime_accel_cpu_bundle_20260406_r2/summary.json" ], "figure_path": "docs/_static/figures/readme_runtime_compare.png", "generated_at_utc": "2026-05-19T09:00:26.703287+00:00", "gpu_summary_paths": [], "host": "Rogerios-MacBook-Pro.local", "platform": "macOS-14.4.1-arm64-arm-64bit", "python": "3.11.14", "table_path": "outputs/readme_runtime_table.md" }, "records": [ { "axisymmetric": true, "boundary": "fixed", "case_id": "ITERModel", "lasym": false, "vmec2000_memory_bytes": 93440832, "vmec2000_runtime_s": 0.5476059159846045, "vmec_jax_cold_runtime_s": 3.274011583009269, "vmec_jax_cold_speedup_vs_vmec2000": 0.1672583929838388, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1088853184, "vmec_jax_warm_runtime_s": 2.287840708013391, "vmec_jax_warm_speedup_vs_vmec2000": 0.23935491403162815, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": false, "boundary": "fixed", "case_id": "LandremanPaul2021_QA_lowres", "lasym": false, "vmec2000_memory_bytes": 90557184, "vmec2000_runtime_s": 5.918600000004517, "vmec_jax_cold_runtime_s": 21.20980858398252, "vmec_jax_cold_speedup_vs_vmec2000": 0.2790501374196369, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1261377728, "vmec_jax_warm_runtime_s": 18.22270329098683, "vmec_jax_warm_speedup_vs_vmec2000": 0.3247926449492227, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": false, "boundary": "fixed", "case_id": "LandremanPaul2021_QA_lowres1", "lasym": false, "vmec2000_memory_bytes": 90573120, "vmec2000_runtime_s": 4.249134042009246, "vmec_jax_cold_runtime_s": 21.518316000001505, "vmec_jax_cold_speedup_vs_vmec2000": 0.1974659188948126, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1215141888, "vmec_jax_warm_runtime_s": 16.735871875018347, "vmec_jax_warm_speedup_vs_vmec2000": 0.25389379613690355, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": false, "boundary": "fixed", "case_id": "LandremanPaul2021_QA_reactorScale_lowres", "lasym": false, "vmec2000_memory_bytes": 90442432, "vmec2000_runtime_s": 8.83699570799945, "vmec_jax_cold_runtime_s": 26.678865583002334, "vmec_jax_cold_speedup_vs_vmec2000": 0.3312358121264978, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1050940864, "vmec_jax_warm_runtime_s": 23.305201667011715, "vmec_jax_warm_speedup_vs_vmec2000": 0.37918554982976743, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": false, "boundary": "fixed", "case_id": "LandremanPaul2021_QH_reactorScale_lowres", "lasym": false, "vmec2000_memory_bytes": 90589760, "vmec2000_runtime_s": 10.477135624998482, "vmec_jax_cold_runtime_s": 34.98014637499, "vmec_jax_cold_speedup_vs_vmec2000": 0.2995166318826325, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1137104448, "vmec_jax_warm_runtime_s": 31.852235375001328, "vmec_jax_warm_speedup_vs_vmec2000": 0.3289293671746906, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "LandremanSengupta2019_section5.4_B2_A80", "lasym": false, "vmec2000_memory_bytes": 92359552, "vmec2000_runtime_s": 0.23428066601627506, "vmec_jax_cold_runtime_s": 4.760488875006558, "vmec_jax_cold_speedup_vs_vmec2000": 0.04921357284254809, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1003541888, "vmec_jax_warm_runtime_s": 1.8183468749921303, "vmec_jax_warm_speedup_vs_vmec2000": 0.12884266981089018, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "LandremanSenguptaPlunk_section5p3_low_res", "lasym": true, "vmec2000_memory_bytes": 90638976, "vmec2000_runtime_s": 0.39103241701377556, "vmec_jax_cold_runtime_s": 5.842511416994967, "vmec_jax_cold_speedup_vs_vmec2000": 0.06692882377196943, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1029412160, "vmec_jax_warm_runtime_s": 1.5457833750115242, "vmec_jax_warm_speedup_vs_vmec2000": 0.2529671513712976, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": false, "boundary": "fixed", "case_id": "basic_non_stellsym_pressure", "lasym": true, "vmec2000_memory_bytes": 91081536, "vmec2000_runtime_s": 1.1000057909986936, "vmec_jax_cold_runtime_s": 9.793892499990761, "vmec_jax_cold_speedup_vs_vmec2000": 0.1123154854925895, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 3078350272, "vmec_jax_warm_runtime_s": 5.164276416006032, "vmec_jax_warm_speedup_vs_vmec2000": 0.21300288799208394, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "circular_tokamak", "lasym": false, "vmec2000_memory_bytes": 94145280, "vmec2000_runtime_s": 0.2322230409772601, "vmec_jax_cold_runtime_s": 7.956475292012328, "vmec_jax_cold_speedup_vs_vmec2000": 0.02918667279849328, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1651350080, "vmec_jax_warm_runtime_s": 2.4730411659984384, "vmec_jax_warm_speedup_vs_vmec2000": 0.09390180971108296, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "circular_tokamak_aspect_100", "lasym": false, "vmec2000_memory_bytes": 94571392, "vmec2000_runtime_s": 1.2334602090122644, "vmec_jax_cold_runtime_s": 4.650104291999014, "vmec_jax_cold_speedup_vs_vmec2000": 0.26525431077633255, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1512953920, "vmec_jax_warm_runtime_s": 0.928849124989938, "vmec_jax_warm_speedup_vs_vmec2000": 1.3279446315090475, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "cth_like_fixed_bdy", "lasym": false, "vmec2000_memory_bytes": 92424832, "vmec2000_runtime_s": 0.411813707993133, "vmec_jax_cold_runtime_s": 6.331759082997451, "vmec_jax_cold_speedup_vs_vmec2000": 0.0650393836207332, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1264703424, "vmec_jax_warm_runtime_s": 2.9028567089990247, "vmec_jax_warm_speedup_vs_vmec2000": 0.14186497966519895, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": false, "boundary": "fixed", "case_id": "nfp4_QH_warm_start", "lasym": false, "vmec2000_memory_bytes": 91474624, "vmec2000_runtime_s": 0.34719370800303295, "vmec_jax_cold_runtime_s": 11.11477387501509, "vmec_jax_cold_speedup_vs_vmec2000": 0.031237136437250426, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 2307104832, "vmec_jax_warm_runtime_s": 3.1947174170054495, "vmec_jax_warm_speedup_vs_vmec2000": 0.1086774392485934, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "purely_toroidal_field", "lasym": false, "vmec2000_memory_bytes": 90835584, "vmec2000_runtime_s": 1.7642406660015695, "vmec_jax_cold_runtime_s": 8.280742834002012, "vmec_jax_cold_speedup_vs_vmec2000": 0.21305343027407203, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1823693632, "vmec_jax_warm_runtime_s": 2.3367791659838986, "vmec_jax_warm_speedup_vs_vmec2000": 0.7549881870239705, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "shaped_tokamak_pressure", "lasym": false, "vmec2000_memory_bytes": 93817600, "vmec2000_runtime_s": 0.43498983298195526, "vmec_jax_cold_runtime_s": 5.521187791979173, "vmec_jax_cold_speedup_vs_vmec2000": 0.0787855529228476, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1084347968, "vmec_jax_warm_runtime_s": 2.3684850840072613, "vmec_jax_warm_speedup_vs_vmec2000": 0.18365740866140143, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "solovev", "lasym": false, "vmec2000_memory_bytes": 92900288, "vmec2000_runtime_s": 0.17102229199372232, "vmec_jax_cold_runtime_s": 3.1619852910225745, "vmec_jax_cold_speedup_vs_vmec2000": 0.054086997962730664, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 785585024, "vmec_jax_warm_runtime_s": 1.1155645420076326, "vmec_jax_warm_speedup_vs_vmec2000": 0.1533056004863161, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null }, { "axisymmetric": true, "boundary": "fixed", "case_id": "up_down_asymmetric_tokamak", "lasym": true, "vmec2000_memory_bytes": 94915648, "vmec2000_runtime_s": 0.47556787499343045, "vmec_jax_cold_runtime_s": 4.051771250000456, "vmec_jax_cold_speedup_vs_vmec2000": 0.11737283416317663, "vmec_jax_gpu_cold_runtime_s": null, "vmec_jax_gpu_cold_speedup_vs_vmec2000": null, "vmec_jax_gpu_memory_bytes": null, "vmec_jax_gpu_warm_runtime_s": null, "vmec_jax_gpu_warm_speedup_vs_cpu_warm": null, "vmec_jax_gpu_warm_speedup_vs_vmec2000": null, "vmec_jax_memory_bytes": 1461557248, "vmec_jax_warm_runtime_s": 2.8882609579886775, "vmec_jax_warm_speedup_vs_vmec2000": 0.1646554386569715, "vmecpp_memory_bytes": null, "vmecpp_runtime_s": null, "vmecpp_speedup_vs_vmec2000": null } ] }