Role OverviewAnnapurna Labs is a fully integrated part of AWS, providing infrastructure for the company's products. This role is for a Software Engineer to lead the development of machine learning tools for AWS Neuron, a software stack for cloud-scale machine learning accelerators.
What You Will Do
Design and implement ML infrastructure platform, build systems for capacity management, workload scheduling, and fleet orchestration across ML accelerators. Work with ML scientists, training infrastructure engineers, hardware teams, and internal customers to ensure seamless ML Accelerator access.
Why It Might Be a Fit
Ideal candidate has experience leading machine learning tool projects, deep knowledge of profiling and optimization, resource management, scheduling, code generation, and has worked on new instruction set architectures.
Requirements
- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture of new and existing systems experience
- Experience programming with at least one software programming language
- Experience taking a leading role in building complex software or computing infrastructure
- Experience with AWS Services including EC2, Lambda, S3, DynamoDB, SQS
- Experience in Kubernetes, Docker or containers ecosystem
- Experience managing full application stacks from the OS up through custom applications
- Experience in Bigdata architecture
- Experience with version control systems and CI/CD pipeline implementation
- Strong proficiency in Go/Java, Python, and Javascript/Typescript
- Application and kernel performance profiling and optimization
- Proficiency in integrated software/hardware performance analysis and optimization
- Experience designing and operating production services
Benefits
- health insurance
- dental insurance
- vision insurance
- prescription insurance
- Basic Life & AD&D insurance
- option for Supplemental life plans
- EAP
- Mental Health Support
- Medical Advice Line
- Flexible Spending Accounts
- Adoption and Surrogacy Reimbursement coverage
- 401(k) matching
- paid time off
- parental leave
]]>