Software Engineer- AI/ML, AWS Neuron Distributed Training - Optimization
Listed on 2026-10-05
-
Software Development
Software Engineer, AI Engineer (Applied/Software)
Lead efforts to optimize distributed training performance on AWS Trainium, focusing on maximizing throughput and model flops utilization. Identify and resolve performance bottlenecks across the software stack, from collective communications to compiler optimizations.
RequirementsRequires 3+ years of professional software development experience and 2+ years of experience in system design or architecture. Proficiency in at least one programming language is required, with a preference for candidates with a Bachelor's degree in Computer Science.
Key SkillsDistributed Training, Performance Optimization, PyTorch, JAX, AWS Neuron, Machine Learning, Compiler Optimization, Kernel Performance, System Architecture, Software Development Life Cycle, Collective Communications, Memory Utilization
Benefits- Health Insurance
- Medical Insurance
- Dental Insurance
- Vision Insurance
- Prescription Insurance
- Basic Life & AD&D Insurance
- Supplemental Life Plans
- EAP
- Mental Health Support
- Medical Advice Line
- Flexible Spending Accounts
- Adoption and Surrogacy Reimbursement
- 401(k) Matching
- Paid Time Off
- Parental Leave
- Sign-on Payments
- Restricted Stock Units (RSUs)
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).