AWS Trainium / AI Infrastructure Engineer

Role name: AWS Trainium / AI Infrastructure Engineer

Work Location: USA Remote)

Start date: Immediate availability.

Background check MANDATORY

Job Summary:

• We are looking for an experienced AI Infrastructure Engineer with expertise in AWS cloud, large-scale AI/ML platforms, and distributed model training. The candidate should have hands-on experience with AWS Trainium/Inferentia, AWS Neuron SDK, SageMaker, EKS/Kubernetes, and scalable AI infrastructure. AWS positions Trainium as a purpose-built accelerator for AI training and inference at scale.

Key Skills:

• Experience with AWS Trainium, Inferentia, EC2 Trn instances, and AWS Neuron SDK for AI model training and inference.
• Strong understanding of LLMs, Generative AI, distributed training, and AI/ML infrastructure. [
• Hands-on experience with Kubernetes (EKS), Docker, Python, PyTorch, and cloud-native architectures.
• Experience in AI performance optimization, benchmarking, and migration from GPU-based environments to AWS AI accelerators.

Experience:

• 5+ years in Cloud, Data, or AI Infrastructure.
• Experience supporting large-scale AI/ML training workloads in AWS environments.
• Exposure to Trainium or Inferentia is highly preferred.