greenhouse
Tech Lead Manager- MLRE, ML Systems
Scale AI
CAN
Lead
Discovered 2 weeks ago
pytorch
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
pytorch
Key Skills for This Role
pytorch
Full Job Posting
You will:
Build, profile and optimize our training and inference framework.
Collaborate with ML and research teams to accelerate their research and development, and enable them to develop the next generation of models and data curation.
Research and integrate state-of-the-art technologies to optimize our ML system.
Ideally you’d have:
Passionate about system optimization
Experience with multi-node LLM training and inference
Experience with developing large-scale distributed ML systems
Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc.
Strong software engineering skills, proficient in frameworks and tools such as CUDA, Pytorch, transformers, flash attention, etc.
Strong written and verbal communication skills to operate in a cross functional team environment.
Nice to haves:
Demonstrated expertise in post-training methods and/or next generation use cases for large language models including instruction tuning, RLHF, tool use, reasoning, agents, and multimodal, etc.
Compensation packages at Scale for eligible roles include base salary, equity, and benefits. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position and may be inclusive of several career levels at Scale; it will be determined during the interview process based on work location and additional factors, including job-related skills, experience, qualifications, interview performance, and relevant education or training. Scale employees in eligible roles are also granted equity based compensation, subject to Board of Director approval. Your recruiter can share more about the specific salary range for your preferred location during the hiring process, and confirm whether the hired role will be eligible for equity grant. You'll also receive benefits including, but not limited to: comprehensive health, dental and vision coverage, retirement benefits, a learning and development stipend, and generous PTO. Additionally, this role may be eligible for additional benefits such as a commuter stipend.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Scale AI
Product Designer, Global Public Sector
Dubai, UAE
MidFull-time
Scale AI is seeking a bilingual Product Designer to own 0-to-1 design for AI-driven products serving government and enterprise clients in the Middle East. The role requires Arabic and English fluency, B2B product experie
Discovered TodayView →
Product Designer, Global Public Sector
, QAT
MidFull-time
Scale AI is seeking a bilingual Product Designer to lead end-to-end design for AI-driven B2B products serving government and enterprise clients across the Middle East. The role requires fluent Arabic and English, experie
Discovered TodayView →
Senior Machine Learning Engineer, Public Sector
, USA
Senior
The goal of a Senior Machine Learning Engineer at Scale is to own how we apply generative AI, agentic AI, computer vision, and reinforcement learning to mission-critical problems in production. Our senior machine learnin
Discovered TodayView →
Software Engineering Intern (Summer 2027)
, QAT
InternInternship
Scale AI is seeking a Software Engineering Intern for Summer 2027 to build and ship AI applications, data pipelines, evaluation infrastructure, and platform tooling. Candidates should be pursuing a relevant bachelor’s de
Discovered 2 days agoView →
Software Engineer - New Grad
, QAT
EntryInternship
Scale AI is seeking a New Grad Software Engineer to build customer-facing AI products, data pipelines, evaluation infrastructure, and reliable platform tooling. Candidates should graduate in Fall 2026 or Spring 2027, hav
Discovered 2 days agoView →
Product Designer, Global Public Sector
Dubai, UAE
Discovered TodayFull-time
Product Designer, Global Public Sector
, QAT
Discovered TodayFull-time
Product Designer, Global Public Sector
Doha, QAT
Discovered Today
Senior Machine Learning Engineer, Public Sector
, USA
Discovered Today
Engineering Manager, Connectivity
New York City, USA
Discovered Yesterday
Software Engineering Intern (Summer 2027)
, QAT
Discovered 2 days agoInternship
Software Engineer - New Grad
, QAT
Discovered 2 days agoInternship
Software Engineer - New Grad
London, GBR
Discovered 2 days ago
