xla

7d6a4f25 - Revive "Modularize zero step function and make it customizable" #7233 (#8332)

Revive "Modularize zero step function and make it customizable" #7233 (#8332) Co-authored-by: Can Karakus <cankrks@gmail.com> Co-authored-by: Karakus <cakarak@amazon.com> Co-authored-by: Shreyas Labhsetwar <137557845+slabhs@users.noreply.github.com>