Overview
Coda to #71/#74 (both merged): measure the compile-time UX exactly as users now get it (cache #128 + autotune-off #132 defaults live), and analyze where any remaining speedup would come.
Plan
- Census (
jax_compile/probe.py): cold (fresh cache) + warm rows for mge & pixelization × jit/vag (+laxmap_vag mge) on local CPU and A100, wrapper defaults only (no manual flags).
- Remaining levers: (a) can
jax.export AOT serialization beat the uncacheable tracing floor (~5–17 s) on repeat runs; (b) do XLA compilation-parallelism flags cut the residual no-autotune compile (MGE vag 117 s CPU / 29 s A100); (c) quantify the warm floor (trace + cache read).
- Deliverable: census table + final addendum in
jax_compile/README.md; follow-ups only if a lever clears the bar.
Prompt: PyAutoMind draft/research/autolens_profiling/final_compile_time_census_and_remaining_speedup.md · branch feature/compile-census-final
Overview
Coda to #71/#74 (both merged): measure the compile-time UX exactly as users now get it (cache #128 + autotune-off #132 defaults live), and analyze where any remaining speedup would come.
Plan
jax_compile/probe.py): cold (fresh cache) + warm rows for mge & pixelization × jit/vag (+laxmap_vag mge) on local CPU and A100, wrapper defaults only (no manual flags).jax.exportAOT serialization beat the uncacheable tracing floor (~5–17 s) on repeat runs; (b) do XLA compilation-parallelism flags cut the residual no-autotune compile (MGE vag 117 s CPU / 29 s A100); (c) quantify the warm floor (trace + cache read).jax_compile/README.md; follow-ups only if a lever clears the bar.Prompt:
PyAutoMind draft/research/autolens_profiling/final_compile_time_census_and_remaining_speedup.md· branchfeature/compile-census-final