[JAX] Estimate post-RHT amax using regular amax#2479

Draft

jberchtold-nvidia wants to merge 1 commit intoNVIDIA:mainfrom

jberchtold-nvidia:jberchtold/estimate-post-rht-amax

Commits on Dec 4, 2025

estimate post-rht amax using regular amax
jberchtold-nvidia
committed