Amazon is filled with workers on visas. At least from the subset I've talked to, this isn't the hill they plan to die on. "Come in or fuck off back to {home country}" is a pretty strong bargaining chip.

For me, getting fired would mean taking a few months off to leisurely reset, then slowly getting back on the leetcode grind, and then trying to find a new remote job. Drawing lines in the sand is much easier when "uproot your entire life" isn't on the table.