Tech titan Amazon is working to step out of Nvidia’s shadow with custom “Trainium” chips designed specially for machine learning as billions of dollars are poured into artificial intelligence (AI).