Rail is not included in Citi's definition of eligible travel purchases:
3% cash back on restaurant (including cafes, bars, lounges and fast food restaurants) and eligible travel purchases worldwide, including airfare, hotels, car rentals, travel agencies, cruise lines and Costco Travel
I wonder if booking Amtrak through a travel agency (if it's even possible these days) would be a workaround to get the 3% benefit.