Senior Machine Learning Systems Engineer, Llm Inference Performance On Aws Trainium (Winnipeg)

Senior Machine Learning Systems Engineer, Llm Inference Performance On Aws Trainium (Winnipeg)

07 Nov
|
Capgemini Engineering
|
Winnipeg

07 Nov

Capgemini Engineering

Winnipeg

At Capgemini Engineering, the world leader in engineering services, we bring together a global team of engineers, scientists, and architects to help the world’s most innovative companies unleash their potential. From autonomous cars to life‑saving robots, our digital and software technology experts think outside the box as they provide unique R&D; and engineering services across all industries.

About The Job You’re ConsideringWe are seeking a hands‑on machine learning systems engineer to optimize production‑grade large language model inference across current and next‑generation AWS Trainium platforms, including Trainium2 and Trainium3. The work focuses on decode latency, throughput under representative load, price‑performance, long‑context and multi‑chip scaling, and reproducible benchmarking against leading GPU platforms. Improve decode latency, throughput, and cost on AWS Trainium platforms, validating correctness at every step.

Initial priorities include speculative decoding and native multi‑token prediction (MTP) for Qwen‑family models in vLLM on AWS Neuron. Deliver latency‑throughput Pareto frontiers and price‑performance comparisons aligned with customer service‑level objectives. Onboard additional architectures, including hybrid state‑space and mixture‑of‑experts models, reusing and generalizing existing implementations where practical.

Create runbooks, implementation guides, and benchmark documentation, and support hands‑on workshops so customer engineers can operate, reproduce, and tune the stack independently. 5+ years of experience in machine learning or systems engineering, including 2+ years focused on production model inference or serving. Strong proficiency in Python and PyTorch, with the ability to read and modify model implementations involving attention, mixture‑of‑experts routing, and KV‑cache management. Hands‑on experience with an LLM serving framework such as vLLM, TensorRT‑LLM, SGLang, or Triton Inference Server.

Deep understanding of LLM inference performance, including prefill and decode, continuous batching, tensor and expert parallelism, and their impact on latency, throughput, and cost.





Experience profiling and optimizing accelerator workloads at the framework, compiler, or kernel level, and designing rigorous performance experiments. Robust written and verbal communication skills, with the ability to explain engineering trade‑offs to technical audiences and business impact to senior stakeholders.

Preferred qualifications include experience with AWS Trainium/Inferentia, the AWS Neuron SDK, and optimizing AI inference workloads, along with expertise in accelerator migrations, speculative decoding techniques, and kernel development using NKI, Triton, or CUDA. Capgemini provides compensation range information in accordance with applicable national, state, provincial, and local pay transparency laws. These may include, but are not limited to: Geographic location, Education and qualifications, Certifications and licenses, Relevant experience and skills, Seniority and performance, Market and business consideration, Internal pay equity.

In addition to base salary, this role may be eligible for additional compensation such as variable incentives, bonuses, or commissions, depending on the position and applicable laws. Capgemini offers a comprehensive, non‑negotiable benefits package to all regular, full‑time employees. S. and Canada, available benefits are determined by local policy and eligibility and may include: Paid time off based on employee grade (A‑F), defined by policy: Vacation: 12-25 days, depending on grade, Company paid holidays, Personal Days, Sick Leave Medical, dental, and vision coverage (or provincial healthcare coordination in Canada) Retirement savings plans (e.Life and disability insurance Employee assistance programs Other benefits as provided by local policy and eligibility Compensation (including bonuses, commissions,



or other forms of incentive pay) is not considered earned, vested, or payable until it becomes due under the terms of applicable plans or agreements and is subject to Capgemini’s discretion, consistent with applicable laws.

Capgemini is an Equal Opportunity Employer encouraging inclusion in the workplace. Capgemini also participates in the Partnership Accreditation in Indigenous Relations (PAIR) program which supports meaningful engagement with Indigenous communities across Canada by promoting fairness, accessibility, inclusion and respect. We value the rich cultural heritage and contributions of Indigenous Peoples and actively work to create a welcoming and respectful environment.

All qualified applicants will receive consideration for employment without regard to race, national origin, gender identity/expression, age, religion, disability, sexual orientation, genetics, veteran status, marital status or any other characteristic protected by law. Physical, mental, sensory or environmental demands may be referenced in an attempt to communicate the manner in which this position traditionally is performed. Whenever necessary to provide individuals with disabilities an equal employment opportunity, Capgemini will consider reasonable accommodations that might involve varying job requirements and/or changing the way this job is performed, provided that such accommodation does not pose an undue hardship.

Please be aware that Capgemini may capture your image (video or screenshot) during the interview process and that image may be used for verification, including during the hiring and onboarding process. Capgemini is a global business and technology transformation partner, helping organizations to accelerate their dual transition to a digital and sustainable world, while creating tangible impact for enterprises and society. It delivers end-to-end services and solutions leveraging strengths from strategy and design to engineering, all fueled by its market leading capabilities in AI, generative AI, cloud and data, combined with its deep industry expertise and partner ecosystem. #

📌 Senior Machine Learning Systems Engineer, Llm Inference Performance On Aws Trainium (Winnipeg)
🏢 Capgemini Engineering
📍 Winnipeg

Reply to this offer

Impress this employer describing Your skills and abilities, fill out the form below and leave Your personal touch in the presentation letter.

Subscribe to this job alert:

Get the latest job offers by email for: senior machine learning systems engineer, llm inference performance on aws trainium (winnipeg) / winnipeg

Subscribe to this job alert:

Get the latest job offers by email for: senior machine learning systems engineer, llm inference performance on aws trainium (winnipeg) / winnipeg