Amazon’s Trainium chip is seeing more traction, and suppliers Astera Labs, Credo and Marvell stand to gain.