Training and inference on AWS Trainium and Inferentia chips. - View it on GitHub