{"id":3683276,"url":"https://alion.io/job/robots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3","title":"Staff ML Engineer – AWS Trainium & SageMaker","company":{"id":7662,"name":"Robots & Pencils","domain":"robotsandpencils.com","url":"https://alion.io/company/robots-and-pencils","size_band":"201-500","is_staffing_agency":false,"employer_type":"services","is_intermediary":false,"listed_via":null,"ats_vendor":"Greenhouse","truth_index":{"grade":"B","score":82,"open_postings":11,"ghost_share":0,"stale_share":0.727,"repost_share":0,"time_to_fill_p50_days":32,"computed_at":"2026-11-05T05:54:30Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"staff","employment_type":null,"work_mode":"remote","remote_scope":"stated_countries","remote_scope_basis":"posting_text","remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["Canada"],"countries":["CA"],"hiring_countries":["CA"],"hiring_countries_total":1,"salary":null,"salary_estimate":null,"experience_years_min":null,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"Amazon SageMaker","optional":false},{"name":"AWS","optional":false},{"name":"AWS Trainium","optional":false},{"name":"Python","optional":false},{"name":"PyTorch","optional":false}],"status":"live","first_seen_at":"2026-11-02T21:10:07Z","employer_posted_date":"2026-11-02","last_verified_at":"2026-11-06T00:50:43Z","board_verified":true,"closed_at":null,"days_open":3,"trust":{"level":"ok","repost_count":null,"flags":[],"days_open":3},"description":"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.\nThe Role\nWe'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.\nPyTorch 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.\nWhat You'll Do\nTrain and operate models on Amazon SageMaker with AWS Trainium as the underlying compute\nWrite 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)\nDiagnose 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\nTranslate a request for \"a Trainium job\" into an actual working, cost-aware training pipeline, end to end\nTune distributed training runs for throughput and cost on SageMaker's training infrastructure\nWork directly with client and internal engineering teams to scope and deliver real production training workloads, not experiments that stay in a notebook\nWhat You'll Bring\nStrong, hands-on PyTorch experience, ideally including distributed or multi-device training\nProduction experience with Amazon SageMaker for training and/or inference\nComfort 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\nAWS 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\nSolid Python fundamentals and comfort operating in a client-facing, production engineering environment","description_format":"text","description_chars":2556,"description_truncated":false,"requirements":{"experience_years_min":null,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":null,"security_clearance":false,"languages":[]},"benefits":[],"hiring_locations":[{"name":"Canada","iso":"CA","kind":"country"}],"hiring_excludes":[],"relocation_offered":false,"industries":["UI/UX Design","Custom Software Development","AI Consulting & Integration"],"lifecycle":[{"event":"open","at":"2026-11-03T13:53:07Z"}],"visa":[],"liveness":{"score":63,"band":"ok","label":"Likely open","p_open":1,"p_active":0.632,"p_room":1,"age_days":2,"expected_fill_days":32,"reasons":["conf:1","stale_co","velocity","win:early"],"computed_at":"2026-11-05T05:54:30Z"},"pay":null,"html_url":"https://alion.io/job/robots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3","json_url":"https://alion.io/job/robots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3.json","meta":{"generated_at":"2026-11-06T02:11:13Z","cache_seconds":300,"methodology":"https://alion.io/methodology","terms":"https://alion.io/terms","contact":"https://alion.io/contact","api":"https://alion.io/developers","about":"Alion is a live layer of people, companies and AI agents: who they are, whether they are real and active right now, what they do and how to work with them, readable by people and by agents and paid per call.","catalog":"https://alion.io/catalog.json","usage":{"tier":"crawler_verified","counted_by":"address","units_charged":1,"used_today":3531,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-11-07T00:00:00Z"}},"offers":[{"id":"company.slices","title":"One company in depth, by slice","status":"live","price":{"credits":0.02,"usd":0.002,"plus_per_slice":{"credits":0.05,"usd":0.005}},"unit":"per company, plus each slice with data","note":"the employer in depth","call":{"mcp_tool":"get_company","arguments":{"id":7662},"rest":"https://alion.io/mcp/rest/get_company?id=7662"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Frobots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3"},{"id":"market.stats","title":"A market slice: pay, demand and time to fill","status":"live","price":{"credits":1,"usd":0.1},"unit":"per slice","note":"pay, demand and time to fill for this role and place","call":{"mcp_tool":"market_stats"},"human":"https://alion.io/catalog?offer=market.stats&for=job%2Frobots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3"},{"id":"job.search","title":"Open jobs by role, technology, place, pay and visa","status":"live","price":{"credits":0.02,"usd":0.002},"unit":"per posting in a list","note":"similar open postings","call":{"mcp_tool":"search_jobs"},"human":"https://alion.io/catalog?offer=job.search&for=job%2Frobots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3"},{"id":"company.verify","title":"Is this company real and active right now","status":"pilot","price":null,"unit":"per company","request":{"url":"https://alion.io/catalog/request","method":"POST","body":"{\"offer\": \"company.verify\", \"for\": \"job/robots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Frobots-pencils-staff-ml-engineer-aws-trainium-sagemaker-3"}],"agent":{"mcp":"https://alion.io/mcp/jobs","mcp_open":"https://alion.io/mcp/open","next":[{"tool":"get_job","arguments":{"id":3683276}},{"tool":"search_jobs","arguments":{"company":"robotsandpencils.com","detail":"compact"}},{"tool":"match_people","arguments":{"job_id":3683276},"note":"people open to offers who fit this vacancy"}],"note":"The same data through MCP tools: https://alion.io/mcp/open needs no sign-in (compact searches, postings, people); https://alion.io/mcp with an Alion account has everything. Docs: https://alion.io/developers"}}