Jobright.ai
ML Acceleration Engineer – Distributed Training, New Grad
Jobright.ai, Cupertino, California, United States, 95014
ML Acceleration Engineer – Distributed Training, New Grad
Join to apply for the
ML Acceleration Engineer – Distributed Training, New Grad
role at
Jobright.ai ML Acceleration Engineer – Distributed Training, New Grad
1 day ago Be among the first 25 applicants Join to apply for the
ML Acceleration Engineer – Distributed Training, New Grad
role at
Jobright.ai Amazon Web Services (AWS) is a leading cloud computing provider, and they are seeking a Machine Learning Acceleration / Framework Engineer to join their Annapurna Labs team. The role involves enhancing machine learning capabilities on AWS's specialized AI hardware by developing distributed training solutions and optimizing models for inference. Responsibilities: • Improving PyTorch and JAX for distributed training on Trainium chips • Optimizing ML models for efficient inference on Inferentia processors • Collaborating with compiler and runtime teams to maximize hardware performance • Developing and integrating new features in ML frameworks to support AWS AI services Qualifications: Required: • To qualify, applicants should have earned (or will earn) a Bachelors or Masters degree between December 2022 and September 2025. • Working knowledge of C++ and Python • Experience with ML frameworks, particularly PyTorch, Jax, and/or vLLM • Understanding of parallel computing concepts and CUDA programming Preferred: • Open source contributions to ML frameworks or tools • Experience optimizing ML workloads for performance • Direct experience with PyTorch internals or CUDA optimization • Hands-on experience with LLM infrastructure tools (e.g., vLLM, TensorRT) 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 Entry 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 Machine Learning Engineer jobs in
Cupertino, CA . AI/ML Engineer (Multiple roles and seniority levels)
San Jose, CA $120,700.00-$228,600.00 4 days ago San Jose, CA $137,500.00-$236,500.00 2 days ago San Jose, CA $120,700.00-$228,600.00 4 days ago New Grads 2025 - Software Engineer - Computer Vision/Deep Learning
San Jose, CA $120,000.00-$165,000.00 9 months ago San Jose, CA $137,500.00-$236,500.00 4 weeks ago San Jose, CA $120,700.00-$228,600.00 2 days ago New Grads 2025 - Software Engineer, Algorithm
San Jose, CA $120,000.00-$165,000.00 10 months ago Sunnyvale, CA $158,200.00-$185,000.00 2 weeks ago San Jose, CA $120,700.00-$228,600.00 4 days ago Machine Learning Engineer (I, II, or Sr.)
Santa Clara, CA $137,300.00-$212,900.00 5 days ago Machine Learning Engineer, Early Stage Project
Mountain View, CA $136,000.00-$185,000.00 5 days ago Sunnyvale, CA $167,000.00-$185,500.00 2 weeks ago Machine Learning Engineer (I, II, or Sr.)
Sunnyvale, CA $167,000.00-$185,500.00 1 month ago Software Engineer, AI Platform - New Grad
Mountain View, CA $145,000.00-$170,000.00 21 hours ago San Jose, CA $120,000.00-$240,000.00 4 months ago Mountain View, CA $141,000.00-$202,000.00 3 days ago San Jose, CA $169,500.00-$291,500.00 2 days ago San Jose, CA $118,657.00-$187,200.00 1 week ago San Jose, CA $130,000.00-$182,000.00 11 months ago New Grads 2025 - General Software Engineer
San Jose, CA $120,000.00-$165,000.00 6 months ago San Jose, CA $120,700.00-$301,200.00 4 days ago San Jose, CA $120,700.00-$228,600.00 4 days ago Redwood City, CA $167,200.00-$250,800.00 1 week ago Menlo Park, CA $180,000.00-$200,000.00 1 month ago San Jose, CA $142,700.00-$257,600.00 4 days 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
ML Acceleration Engineer – Distributed Training, New Grad
role at
Jobright.ai ML Acceleration Engineer – Distributed Training, New Grad
1 day ago Be among the first 25 applicants Join to apply for the
ML Acceleration Engineer – Distributed Training, New Grad
role at
Jobright.ai Amazon Web Services (AWS) is a leading cloud computing provider, and they are seeking a Machine Learning Acceleration / Framework Engineer to join their Annapurna Labs team. The role involves enhancing machine learning capabilities on AWS's specialized AI hardware by developing distributed training solutions and optimizing models for inference. Responsibilities: • Improving PyTorch and JAX for distributed training on Trainium chips • Optimizing ML models for efficient inference on Inferentia processors • Collaborating with compiler and runtime teams to maximize hardware performance • Developing and integrating new features in ML frameworks to support AWS AI services Qualifications: Required: • To qualify, applicants should have earned (or will earn) a Bachelors or Masters degree between December 2022 and September 2025. • Working knowledge of C++ and Python • Experience with ML frameworks, particularly PyTorch, Jax, and/or vLLM • Understanding of parallel computing concepts and CUDA programming Preferred: • Open source contributions to ML frameworks or tools • Experience optimizing ML workloads for performance • Direct experience with PyTorch internals or CUDA optimization • Hands-on experience with LLM infrastructure tools (e.g., vLLM, TensorRT) 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 Entry 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 Machine Learning Engineer jobs in
Cupertino, CA . AI/ML Engineer (Multiple roles and seniority levels)
San Jose, CA $120,700.00-$228,600.00 4 days ago San Jose, CA $137,500.00-$236,500.00 2 days ago San Jose, CA $120,700.00-$228,600.00 4 days ago New Grads 2025 - Software Engineer - Computer Vision/Deep Learning
San Jose, CA $120,000.00-$165,000.00 9 months ago San Jose, CA $137,500.00-$236,500.00 4 weeks ago San Jose, CA $120,700.00-$228,600.00 2 days ago New Grads 2025 - Software Engineer, Algorithm
San Jose, CA $120,000.00-$165,000.00 10 months ago Sunnyvale, CA $158,200.00-$185,000.00 2 weeks ago San Jose, CA $120,700.00-$228,600.00 4 days ago Machine Learning Engineer (I, II, or Sr.)
Santa Clara, CA $137,300.00-$212,900.00 5 days ago Machine Learning Engineer, Early Stage Project
Mountain View, CA $136,000.00-$185,000.00 5 days ago Sunnyvale, CA $167,000.00-$185,500.00 2 weeks ago Machine Learning Engineer (I, II, or Sr.)
Sunnyvale, CA $167,000.00-$185,500.00 1 month ago Software Engineer, AI Platform - New Grad
Mountain View, CA $145,000.00-$170,000.00 21 hours ago San Jose, CA $120,000.00-$240,000.00 4 months ago Mountain View, CA $141,000.00-$202,000.00 3 days ago San Jose, CA $169,500.00-$291,500.00 2 days ago San Jose, CA $118,657.00-$187,200.00 1 week ago San Jose, CA $130,000.00-$182,000.00 11 months ago New Grads 2025 - General Software Engineer
San Jose, CA $120,000.00-$165,000.00 6 months ago San Jose, CA $120,700.00-$301,200.00 4 days ago San Jose, CA $120,700.00-$228,600.00 4 days ago Redwood City, CA $167,200.00-$250,800.00 1 week ago Menlo Park, CA $180,000.00-$200,000.00 1 month ago San Jose, CA $142,700.00-$257,600.00 4 days 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