Jobright.ai
Software Engineer – AI/ML, AWS Neuron, Mid-Level
Jobright.ai - Cupertino, California, United States, 95014
Work at Jobright.ai
Overview
- View job
Overview
Software Engineer – AI/ML, AWS Neuron, Mid-Level
Join to apply for the
Software Engineer – AI/ML, AWS Neuron, Mid-Level
role at
Jobright.ai Software Engineer – AI/ML, AWS Neuron, Mid-Level
3 days ago Be among the first 25 applicants Join to apply for the
Software Engineer – AI/ML, AWS Neuron, Mid-Level
role at
Jobright.ai Get AI-powered advice on this job and more exclusive features. Amazon Web Services (AWS) is seeking a Software Development Engineer II to join the Annapurna Labs team, focusing on building and maintaining complex software products that enhance customer experiences. This role involves developing distributed training solutions for machine learning models, collaborating with various engineering disciplines to ensure optimal performance on AWS Trainium and Inferentia silicon. Responsibilities: • Help lead the efforts building distributed training support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. • Help tune these models to ensure highest performance and maximize the efficiency of them running on the customer AWS Trainium and Inferentia silicon and the TRn1 , Inf1 servers. Qualifications: Required: • 3+ years of non-internship professional software development experience • 3+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience • Experience programming with at least one software programming language • Deep Learning industry experience Preferred: • 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience • Bachelor's degree in computer science or equivalent • Preferred previous software engineer expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training. The group presents lot of opportunity for optimization and scaling large deep learning models on Trainium architecture. Company: Launched in 2006, Amazon Web Services (AWS) began exposing key infrastructure services to businesses in the form of web services -- now widely known as cloud computing. Founded in 2002, the company is headquartered in Seattle, Washington, USA, with a team of 10001+ employees. The company is currently Late Stage. Amazon Web Services (AWS) has a track record of offering H1B sponsorships. Seniority level
Seniority level Mid-Senior level Employment type
Employment type Full-time Job function
Industries Software Development Referrals increase your chances of interviewing at Jobright.ai by 2x Inferred from the description for this job
Medical insurance Vision insurance 401(k) Get notified about new Software Engineer jobs in
Cupertino, CA . Palo Alto, CA $160,000.00-$180,000.00 2 weeks ago Mountain View, CA $145,000.00-$170,000.00 1 week ago Mountain View, CA $138,225.00-$207,575.00 1 week ago Software Engineer, AI Platform - New Grad
Mountain View, CA $145,000.00-$170,000.00 1 week ago Software Engineer (L4), Content & Business Products
San Francisco Bay Area $160,000.00-$180,000.00 2 weeks ago Mountain View, CA $130,000.00-$176,000.00 4 days ago Palo Alto, CA $96,000.00-$200,000.00 2 weeks ago San Jose, CA $100,500.00-$173,250.00 2 days ago San Jose, CA $113,400.00-$206,300.00 1 week ago Software Engineer (L5) - Open Connect Platform
San Jose, CA $113,400.00-$206,300.00 1 week ago Palo Alto, CA $107,000.00-$145,400.00 2 weeks ago San Jose, CA $113,400.00-$206,300.00 1 week ago Palo Alto, CA $115,000.00-$260,000.00 2 weeks ago San Jose, CA $100,500.00-$173,250.00 3 weeks ago eCommerce Full Stack Developer (React / Shopify) - On Site
San Jose, CA $93,200.00-$170,600.00 1 week ago Software Engineer(s) - New Grad (Fall 2025 Graduation)
San Jose, CA $113,400.00-$206,300.00 1 week ago Menlo Park, CA $157,000.00-$230,000.00 2 weeks ago San Jose, CA $113,400.00-$206,300.00 1 week ago Software Development Engineer - Front End
San Jose, CA $133,900.00-$242,000.00 4 days ago San Jose, CA $113,400.00-$206,300.00 1 week ago We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.
#J-18808-Ljbffr
Join to apply for the
Software Engineer – AI/ML, AWS Neuron, Mid-Level
role at
Jobright.ai Software Engineer – AI/ML, AWS Neuron, Mid-Level
3 days ago Be among the first 25 applicants Join to apply for the
Software Engineer – AI/ML, AWS Neuron, Mid-Level
role at
Jobright.ai Get AI-powered advice on this job and more exclusive features. Amazon Web Services (AWS) is seeking a Software Development Engineer II to join the Annapurna Labs team, focusing on building and maintaining complex software products that enhance customer experiences. This role involves developing distributed training solutions for machine learning models, collaborating with various engineering disciplines to ensure optimal performance on AWS Trainium and Inferentia silicon. Responsibilities: • Help lead the efforts building distributed training support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. • Help tune these models to ensure highest performance and maximize the efficiency of them running on the customer AWS Trainium and Inferentia silicon and the TRn1 , Inf1 servers. Qualifications: Required: • 3+ years of non-internship professional software development experience • 3+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience • Experience programming with at least one software programming language • Deep Learning industry experience Preferred: • 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience • Bachelor's degree in computer science or equivalent • Preferred previous software engineer expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training. The group presents lot of opportunity for optimization and scaling large deep learning models on Trainium architecture. Company: Launched in 2006, Amazon Web Services (AWS) began exposing key infrastructure services to businesses in the form of web services -- now widely known as cloud computing. Founded in 2002, the company is headquartered in Seattle, Washington, USA, with a team of 10001+ employees. The company is currently Late Stage. Amazon Web Services (AWS) has a track record of offering H1B sponsorships. Seniority level
Seniority level Mid-Senior level Employment type
Employment type Full-time Job function
Industries Software Development Referrals increase your chances of interviewing at Jobright.ai by 2x Inferred from the description for this job
Medical insurance Vision insurance 401(k) Get notified about new Software Engineer jobs in
Cupertino, CA . Palo Alto, CA $160,000.00-$180,000.00 2 weeks ago Mountain View, CA $145,000.00-$170,000.00 1 week ago Mountain View, CA $138,225.00-$207,575.00 1 week ago Software Engineer, AI Platform - New Grad
Mountain View, CA $145,000.00-$170,000.00 1 week ago Software Engineer (L4), Content & Business Products
San Francisco Bay Area $160,000.00-$180,000.00 2 weeks ago Mountain View, CA $130,000.00-$176,000.00 4 days ago Palo Alto, CA $96,000.00-$200,000.00 2 weeks ago San Jose, CA $100,500.00-$173,250.00 2 days ago San Jose, CA $113,400.00-$206,300.00 1 week ago Software Engineer (L5) - Open Connect Platform
San Jose, CA $113,400.00-$206,300.00 1 week ago Palo Alto, CA $107,000.00-$145,400.00 2 weeks ago San Jose, CA $113,400.00-$206,300.00 1 week ago Palo Alto, CA $115,000.00-$260,000.00 2 weeks ago San Jose, CA $100,500.00-$173,250.00 3 weeks ago eCommerce Full Stack Developer (React / Shopify) - On Site
San Jose, CA $93,200.00-$170,600.00 1 week ago Software Engineer(s) - New Grad (Fall 2025 Graduation)
San Jose, CA $113,400.00-$206,300.00 1 week ago Menlo Park, CA $157,000.00-$230,000.00 2 weeks ago San Jose, CA $113,400.00-$206,300.00 1 week ago Software Development Engineer - Front End
San Jose, CA $133,900.00-$242,000.00 4 days ago San Jose, CA $113,400.00-$206,300.00 1 week ago We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.
#J-18808-Ljbffr