Member of Technical Staff – Model Training
Requirements
- Education
- Bachelor's degree
- Visa
- Sponsorship available
Benefits
Joblaze summary
In the role of Model Training engineer at Inflection AI, the individual will focus on developing and optimizing post-training pipelines to enhance the performance of large language models. Proficiency in PyTorch and experience with techniques like RLHF and DPO are essential, as the position requires hands-on work with distributed training and fine-tuning. This role is well-suited for someone with a strong technical background and a balance of research and practical application, particularly in AI model training. Inflection AI emphasizes collaboration across teams to ensure improvements are effectively integrated into their AI products.
Joblaze insights
- Listed about a month ago on Joblaze — tracked directly from Inflection AI's career page.
- Inflection AI has 3 other open roles including Research Engineer, Voice, Senior Backend Engineer, Agentic, Head of IT & Internal Tools.
- 160 active PyTorch roles on Joblaze right now.
- 14 active RLHF roles on Joblaze right now.
- 11 active DeepSpeed roles on Joblaze right now.
- Salary band is in line with the typical range for AI/ML (median ~$190,000).
Quick facts
- Is the Member of Technical Staff – Model Training role remote?
- It's hybrid — Inflection AI expects some on-site time in Palo Alto, CA.
- What's the salary range?
- Inflection AI lists $175,000–$350,000 for this role.
- Where is the role based?
- Inflection AI is hiring for this position in Palo Alto, CA.
- What's the tech stack?
- Joblaze extracted these technologies from the posting: RLAIF, DeepSpeed, DPO, PyTorch, GRPO, RLHF.
- Does Inflection AI sponsor work visas for this role?
- Yes — the posting indicates visa sponsorship is available for the right candidate.
- What seniority level is this role?
- Inflection AI targets mid-level candidates for this position.
- Is this full-time or contract?
- Full-time for this Member of Technical Staff – Model Training role at Inflection AI.
From the original posting
About Inflection AI
Inflection AI is a Public Benefit Corporation empowering people with human-centered, emotionally intelligent AI. We’re shaping the future of AI by combining emotional intelligence (EQ) and raw intelligence (IQ) to elevate people’s potential.
Inflection AI created Pi, the world’s first emotionally intelligent AI, to help people work through decisions, emotions, and challenges. Pi is a personal AI agent powered by Inflection AI’s foundation model, proving that AI can be personal, empathetic, and contextually aware.
About the Role
As a Model Training engineer, you will design, build, and scale the post-training pipelines that turn a general LLM into a brand-fluent, production-ready assistant. Your innovations in fine-tuning and preference optimization (RLHF, DPO, GRPO, RLAIF) will directly improve reliability, alignment, and cost.
This is a good role for you if you:
- Have hands-on experience training and fine-tuning large transformer models on multi-GPU / multi-node clusters.
- Are fluent in PyTorch and its ecosystem tools (Torchtune, FSDP, DeepSpeed) and enjoy digging into distributed-training internals, mixed precision, and memory-efficiency tricks.
- Have shipped or published work in RLHF, DPO, GRPO, or RLAIF and understand their practical trade-offs.
- Care deeply about training tools, pipelines, and reproducibility—you automate the boring parts so you can iterate on the fun parts.
- Balance research curiosity with product pragmatism—you know when to run an ablation and when to ship.
- Communicate crisply with both technical and non-technical teammates.
- Have a bachelor’s degree or equivalent in a related field to the offered position requirements.
Responsibilities include:
- Contribute to end-to-end post-training workflows—dataset curation, hyper-parameter search, evaluation, and rollout—using PyTorch, Torchtune, FSDP/DeepSpeed, and our internal orchestration stack.
- Prototype and compare alignment techniques (e.g., curriculum RL, multi-objective reward modeling, tool-use fine-tuning) and push the best ideas into production.
- Automate training at scale: build robust pipeline components, tools, scripts, and dashboards so experiments are reproducible and easy to trace.
- Define the metrics that matter; run A/B tests and iterate quickly to meet aggressive quality targets.
- Collaborate with inference, safety, and product teams to land improvements in customer-facing systems.
Employee Pay Disclosures
At Inflection AI, we aim to attract and retain the best employees and compensate them in a way that appropriately and fairly values their individual contributions to the company. For this role, Inflection AI estimates a starting annual base salary to fall within the range of $175,000 to $350,000, depending on a candidate’s qualifications and level of experience. This role also includes a meaningful equity component, allowing employees to share in the long-term success of the company.
Benefits
Inflection AI values and supports our team’s mental and physical health. We are focused on building a positive, safe, inclusive and inspiring place to work. Our benefits include:
- Diverse medical, dental and vision options
- 401k matching program
- Unlimited paid time off
- Parental leave and flexibility for all parents and caregivers
- Support of country-specific visa needs for international employees living in the Bay Area