AI Infrastructure Engineer

Job summary

Build and support scalable AWS AI infrastructure for large-scale AI/ML training and inference workloads. Optimize and benchmark model performance, including migrating workloads from GPU-based environments to AWS AI accelerators.

Job details

Role name: AI Infrastructure Engineer Work Location: Canadá (Remote) 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.

What you’ll do

Build and support scalable AWS AI infrastructure for large-scale AI/ML training and inference workloads. Optimize and benchmark model performance, including migrating workloads from GPU-based environments to AWS AI accelerators.

Requirements

Requires 5+ years of experience in cloud, data, or AI infrastructure and experience supporting large-scale AI/ML training workloads in AWS. Candidates should have hands-on experience with AWS AI accelerators and related tools, Kubernetes, Docker, Python, and PyTorch; Trainium or Inferentia experience is highly preferred.

Listed skills

• Kubernetes · Preferred
• Docker · Preferred
• Python · Preferred

Other relevant skills

Identified from the job description. Confirm important requirements above.

• AWS Trainium
• AWS Inferentia
• AWS Neuron SDK
• Amazon SageMaker
• Amazon EKS
• Kubernetes
• Docker
• Python
• PyTorch
• Large Language Models
• Generative AI
• Distributed Training
• AI/ML Infrastructure
• Performance Optimization
• Benchmarking
• Cloud-Native Architectures

Job areas

• Technology
• Software
• Engineering
• Data & Analytics

More jobs from Epsilon Solutions Ltd.