jax

01ed6631 - Add a comment at the end of autotuning a Triton function

Add a comment at the end of autotuning a Triton function I'd like to see what config was chosen at the end of autotuning. Tracking `is the new best config` is a bit hard. PiperOrigin-RevId: 538465212