Train and operate machine learning models on Amazon SageMaker using AWS Trainium. Develop and optimize PyTorch training code for NeuronCore hardware, diagnose device and compiler issues, tune distributed workloads for throughput and cost, and deliver production training pipelines while collaborating directly with client and engineering teams.
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
Robots & Pencils Calgary, Alberta, CAN Office
1507 14 Ave SW, Calgary, Alberta, Canada, T3C 0W4
Similar Jobs
Big Data • Fintech • Mobile • Payments • Financial Services
Own and improve leadership development programs for aspiring, first-time, and newly hired people leaders. Design scalable learning journeys, workshops, peer communities, coaching, assessments, and practical leadership resources. Partner with People teams to connect development with hiring, onboarding, promotions, and growth. Use data, feedback, and emerging technologies to measure impact and continuously improve programs while managing vendors, communications, resources, and budgets.
Top Skills:
Ai Tools
Cloud • Fintech • Food • Information Technology • Software • Hospitality
Prospect restaurants, manage the full sales cycle, conduct product demonstrations, develop customized solutions, negotiate with decision-makers, and close new accounts across the North York/North Toronto territory. The role requires maintaining sales activity in Salesforce, collaborating with internal teams to ensure successful delivery, analyzing competitors, and meeting sales goals. This is a field-based position requiring local residency or relocation and reliable transportation.
Top Skills:
CRMSalesforce
An Hour Ago
Artificial Intelligence • Healthtech • Machine Learning • Natural Language Processing • Biotech • Pharmaceutical
Promotes Pfizer specialty care and internal medicine products to healthcare professionals across Montérégie. The representative manages territory sales objectives, develops business plans, builds customer relationships, conducts face-to-face and virtual engagements, coordinates continuing medical education programs and congress activities, reports adverse events and product complaints, and maintains company assets. The role requires pharmaceutical sales expertise, bilingual French and English communication, digital and AI tool proficiency, regulatory compliance, and a valid driver’s license.
Top Skills:
Artificial IntelligenceComputer SoftwareDigital ToolsMicrosoft Copilot
What you need to know about the Calgary Tech Scene
Employees can spend up to one-third of their life at work, so choosing the right company is crucial, not just for the job itself but for the company culture as well. While startups often offer dynamic culture and growth opportunities, large corporations provide benefits like career development and networking, especially appealing to recent graduates. Fortunately, Calgary stands out as a hub for both, recognized as one of Startup Genome's Top 100 Emerging Ecosystems, while also playing host to a number of multinational enterprises. In Calgary, job seekers can find a wide range of opportunities.
