✦ Luna Orbit — AI & Machine Learning

Software Development Engineer, Neuron Foundation Tools

at Amazon.com

📍 US, WA, Seattle Unknown Posted March 13, 2026
Apply Now → ← Browse All Jobs

Position Details

Type Full-Time
Experience mid
Exp. Years 3+ years
Education Not specified
Category AI & Machine Learning

About this role

Join AWS as an AI/ML Engineer focusing on performance profiling tools for AI accelerators, optimizing ML workloads, and developing high-performance system tools.

Key Responsibilities

  • Develop performance profiling tools
  • Optimize ML workloads
  • Collaborate on system performance analysis
  • Work with AI accelerators
  • Improve system bottleneck detection

Technical Overview

Work with C++, PyTorch, TensorFlow, and XLA to develop profiling and performance analysis tools for AI hardware accelerators, improving system efficiency and workload performance.

Ideal Candidate

The ideal candidate is a mid-level AI/ML engineer with 3+ years experience in performance analysis and profiling, proficient in C++ and ML frameworks like PyTorch and TensorFlow. They are innovative problem-solvers focused on optimizing AI workloads on hardware accelerators.

Must-Have Skills

Experience with ML profiling toolsExperience with AI/ML performance analysisExperience with C++ developmentExperience with ML frameworks like PyTorchTensorFlow

Nice-to-Have Skills

Experience with ML acceleratorsExperience with performance bottleneck analysisExperience with customer-facing tools

Tools & Platforms

PyTorchTensorFlowXLAC++

Required Skills

C++PyTorchTensorFlowXLAML frameworksPerformance profilingAI acceleratorsNeural networksML workloadsSystem optimization

Hard Skills

C++PyTorchTensorFlowXLAML frameworksPerformance profilingAI acceleratorsNeural networksML workloadsSystem optimization

Soft Skills

collaborationproblem-solvinginnovationcustomer focusteamwork

Industry & Role

Industry Technology/Cloud/AI
Job Function AI/ML performance analysis and system optimization for hardware accelerators

Keywords for Your Resume

C++PyTorchTensorFlowXLAML frameworksPerformance profilingAI acceleratorsNeural networksML workloadsSystem optimization

Deal Breakers

Lack of experience with ML profiling tools, No experience with C++ or ML frameworks, Insufficient background in performance analysis

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile