Software Engineer- AI/ML, Amazon Neuron Training
Seattle, Washington, USA
USD 158K-223K (estimate) Mid-level Full Time
Tasks
- Apply reduced precision formats
- Contribute upstream to open source frameworks
- Develop and tune parallelism strategies
- Enable and optimize large scale training workloads
- Improve training throughput across Neuron software stack
- Optimize distributed training performance on Trainium
- Profile workloads to identify compute memory collectives and host bottlenecks
- Translate performance gaps into framework requirements
Perks/Benefits
- 401k matching
- Dental insurance
- Employee assistance program
- Flexible work hours
- Health insurance
- Medical Advice Line
- Mental health support
- Paid time off
- Parental leave
- Vision insurance
- Work-life balance
Skills/Tech-stack
Build Automation | Code review | Collectives Library | Compiler optimization | Context Parallelism | Data parallelism | Distributed Systems | Expert parallelism | High Performance | High-Performance Computing | JAX | Machine Learning | Parallel Computing | Performance Computing | Pipeline parallelism | Profiling | PyTorch | Python | Reduced precision | Runtime optimization | Source Control | Tensor Parallelism | Testing
Education
Related jobs
-
Senior AI Engineer - Finance USD 140K-203KAWS | Azure | CI/CD | Cloud platform | Data EngineeringDental insurance | Disability insurance | Medical insurance | Paid time off | Supplemental insuranceSenior-level Full TimeWashington, DC, US9h ago
-
Senior Software Engineer - AI Interaction Evaluator (Codex / Claude Code, up to $200/hr) USD 200K-400KClaude Code | Cursor | JavaScript | OpenAI Codex | PythonSenior-level Full TimeMiami R12h ago
-
AI Security Engineer USD 140K-200K800-53 | ABAC | AI RMF | AI Studio | AKS401k matching | Dental insurance | Employee development program | Employee referral program | Medical insuranceSenior-level Full TimeRemote, MD, US R12h ago
-
AWS | Agentic AI | Amazon Bedrock | Amazon SageMaker | Amazon SageMaker AISenior-level Full TimeEast Palo Alto, California, USA14h ago
-
Senior Specialist Solutions Architect - Agentic AI, AGS Specialist Solutions Architects USD 153K-207KAI Governance | AWS | Agent evaluation | Amazon AgentCore | Amazon BedrockSenior-level Full TimeSeattle, Washington, USA14h ago
-
Senior AI Engineer - Agentic USD 123K-185K800-53 | APIs | Agent Orchestration | Agent to Agent | Agent to Agent A2ASenior-level Full TimeNon-Campus Other, United States14h ago
-
Senior AI Engineer – Physical USD 123K-185KComputer Vision | Cyber-Physical Systems | Deep learning | Distributed Systems | Edge ComputingSenior-level Full TimeNon-Campus Other, United States14h ago
-
Senior AI Engineer - Platform USD 123K-185K800-53 | API Development | Access Control | Containerization | CybersecuritySenior-level Full TimeNon-Campus Other, United States14h ago
-
Senior Middleware Software Development Engineer USD 128K-245KC# | C++ | Distributed Computing | GPU Software Development | GPU softwareHealth benefits | Hybrid work model | Retirement benefits | VacationSenior-level Full TimeUSA - OR - Hillsboro, United … R14h ago
-
Junior AI Engineer USD 86K-120KAPI Integration | Authentication | Cloud deployment | Docker | Error Handling401k matching | BYOD reimbursement | Company holidays | Flexible spending accounts | Health savings accountEntry-level Full TimeHouston, Texas, United States (Hybrid) R14h ago
-
Circuit analysis | Communication Systems | Electrical Engineering | Embedded Systems | LinuxFlexible working hours | In-person workEntry-level Internship Part TimeIrvine, CA14h ago
-
Solution Engineer, Enterprise Acquisition USD 165K-216KData Warehousing | Generative AI | Language Models | Large Language Models | Machine LearningSenior-level Full TimeUS-NY-New York14h ago
-
Data Infrastructure Engineer (Query Engine) USD 185K-250KApache Spark | Avro | C++ | Caching | Concurrency401k | Dental insurance | Flexible work hours | Health insurance | Immigration assistanceMid-level Full TimeSan Mateo, CA14h ago
-
Staff Machine Learning Engineer, Traffic Intelligence USD 212K-265KBayesian Calibration | Bayesian Inference | Calibration | Causal Inference | Data PipelinesSenior-level Full TimeUnited States R14h ago
-
Senior Machine Learning Engineer USD 150K-241KC++ | Cloud Architecture | Computer Vision | Data Privacy | EncryptionDiscretionary paid time off | Emotional & mental wellness support | Employer 401k match | Fitness programs | Learning and development programsSenior-level Full TimeSeattle, Washington, United States R14h ago
-
Active Learning | Auto-labeling | Computer Vision | Confidence Scores | Data BalancingBonus program | Company benefits program | Equity incentive planSenior-level Full TimeMountain View, CA, USA; San Francisco, … R15h ago
-
Consultant – Compliance Data Science & AI USD 98K-122KAWS | Anthropic | Azure | Data Pipelines | Data Visualization401k match | Cell phone stipend | Employee assistance program | FSA | HSAMid-level Full TimeSan Francisco, CA, United States15h ago
-
Associate Consultant, Generative AI USD 84K-122KAWS | Azure | CI/CD | Cloud platform | Compliance401k match | Employee assistance program | Flexible spending account | Health savings account | Paid parental leaveMid-level Full TimeSan Francisco, CA, United States15h ago
-
Consultant- Marketing Data Science & AI USD 95K-116KAWS | Azure | Cloud platform | Data Engineering | Data Visualization401k match | Generous PTO | Medical, dental, and vision | Paid parental leaveMid-level Full TimeNew York, NY, United States15h ago
-
Associate Consultant, Generative AI USD 80K-110KAWS | Azure | CI/CD | Docker | Evaluation401k matching | Cell phone stipend | College Savings Assistance | Dental insurance | Employee assistance programMid-level Full TimeNew York, NY, United States15h ago
-
AI & Data Engineer USD 140K-188KAEM Sites | AWS | AWS Bedrock | Adobe Experience | Adobe Experience CloudSenior-level Full TimeLawrence Township, New Jersey, United States15h ago
-
Senior-level Full TimeMountain View, CA15h ago
-
Senior Systems Engineer, Database USD 170K-250KACID | AWS Lambda | Algorithms | Amazon S3 | Amazon Web ServicesIn-office workSenior-level Full TimeVienna, VA, USA16h ago
-
Senior AI/ML Architect, Applied Field Engineering USD 165K-216KAWS | Apache Airflow | Apache Spark | As-a-Service | Cloud platformSenior-level Full TimeUS-NY-New York16h ago
-
Senior Software Engineer, Data & Model USD 150K-300KCloud infrastructure | Containers | Data Pipelines | Dataset Management | Distributed ComputingSenior-level Full TimeFremont Office16h ago