Train and operate production machine learning models on Amazon SageMaker using AWS Trainium. Optimize PyTorch code for Trainium compilation, memory, throughput, and distributed training; diagnose hardware, compiler, device, data, and code issues; build cost-aware end-to-end training pipelines; and collaborate directly with client and internal engineering teams to deliver production workloads.
Robots & Pencils is an AWS Partner building production AI systems for enterprise clients who need real engineering, not proofs of concept that never ship. We work forward-deployed, embedded directly with client teams, solving the problems that are too new or too specialized for a typical vendor relationship to handle.
The Role
We're looking for an engineer who can operate and train models on Amazon SageMaker running on AWS Trainium, AWS's custom silicon built specifically for large-scale model training. This isn't a role where you call an API and wait. You'll be walking up the stack: understanding what a training request actually looks like at the Trainium hardware and compiler level, then carrying that understanding all the way up through PyTorch training code and into a production SageMaker pipeline.
PyTorch is the backbone of this work. If you know the framework deeply and you're comfortable reasoning about how your code actually behaves on custom accelerator hardware rather than treating it as a black box, this role is built around that skill set specifically.
What You'll Do
- Train and operate models on Amazon SageMaker with AWS Trainium as the underlying compute
- Write and optimize PyTorch training code with a real understanding of how it compiles and executes on Trainium (NeuronCore architecture, compiler behavior, memory and throughput tradeoffs)
- Diagnose training run issues that show up specifically because of the hardware, not just the model, distinguishing a data or code problem from a compiler or device-level one
- Translate a request for "a Trainium job" into an actual working, cost-aware training pipeline, end to end
- Tune distributed training runs for throughput and cost on SageMaker's training infrastructure
- Work directly with client and internal engineering teams to scope and deliver real production training workloads, not experiments that stay in a notebook
What You'll Bring
- Strong, hands-on PyTorch experience, ideally including distributed or multi-device training
- Production experience with Amazon SageMaker for training and/or inference
- Comfort working close to the hardware layer: you understand device-specific compilation and can debug issues that are actually about the accelerator, not just the model
- AWS Trainium or Inferentia (Neuron SDK) experience is a strong plus; if you don't have it yet but have deep PyTorch and a track record of picking up new hardware targets fast, we want to talk to you
- Solid Python fundamentals and comfort operating in a client-facing, production engineering environment
Similar Jobs
Cloud • Greentech • Social Impact • Software • Consulting
Own and grow a portfolio of enterprise customers by identifying expansion opportunities, selling new modules and users, coordinating cross-functional teams, maintaining account plans and forecasts, engaging executives, and closing multi-stakeholder deals with 10-20% travel.
Top Skills:
6SenseG2GongLinkedin Sales NavigatorOutreachSalesforceZoominfo
Cloud • Greentech • Social Impact • Software • Consulting
Own strategy, execution, budgets, optimization, testing, tracking, attribution, and reporting for paid search, YouTube, and emerging AI advertising channels. Drive qualified B2B SaaS demand and pipeline across complex campaigns, using performance data to guide investment decisions. Collaborate with lifecycle, product marketing, creative, web, analytics, and marketing operations teams to improve landing pages, lead quality, nurture alignment, and conversion. Present performance insights and recommendations to senior leadership.
Top Skills:
6SenseBing AdsChatgptEloquaGa4Google AdsGoogle Tag ManagerLooker StudioMarketoMicrosoft AdsPardotPower BISalesforceYoutube Ads
Cloud • Greentech • Social Impact • Software • Consulting
Manage a portfolio of global enterprise customers to drive expansion and maximize bookings. Lead strategic, outcome-focused selling, coordinate cross-functional teams, manage RFPs/procurement/legal reviews, maintain account plans and forecasting, and engage executives. Expect 10–20% travel for in-person strategic meetings.
Top Skills:
6SenseChallengerG2GongLinkedin Sales NavigatorMeddpiccOutreachRichardsonSalesforceSpinZoominfo
What you need to know about the Toronto Tech Scene
Although home to some of the biggest names in tech, including Google, Microsoft and Amazon, Toronto has established itself as one of the largest startup ecosystems in the world. And with over 2,000 startups — more than 30 percent of the country's total startups — Toronto continues to attract new businesses. Be it helping entrepreneurs manage their finances, simplifying business operations by automating payroll or assisting pharmaceutical companies in launching new drugs, the city's tech scene is just getting started.
