Member of Technical Staff – Model Training
Inflection AIHighlight
Summary
About the Role
About You
- Hands-on experience training and fine-tuning large transformer models on multi-GPU / multi-node clusters.
- Fluency in PyTorch and its ecosystem tools (Torchtune, FSDP, DeepSpeed).
- A bachelor’s degree or equivalent in a related field.
- Experience with RLHF, DPO, GRPO, or RLAIF and understanding their practical trade-offs.
- Strong communication skills with both technical and non-technical teammates.
Benefits
- Salary Range: $175k – $350k USD per year based on experience and location.
- Competitive stock options.
- Diverse medical, dental, and vision options.
- 401k matching program.
- Unlimited paid time off.
- Parental leave and flexibility for all parents and caregivers.
- Support for country-specific visa needs for international employees.
Inflection AI is a public benefit corporation leveraging our world class large language model to build the first AI platform focused on the needs of the enterprise. Inflection AI was re-founded in March of 2024 and our leadership team has assembled a team of kind, innovative, and collaborative individuals focused on building enterprise AI solutions. We are an organization passionate about what we are building, enjoy working together and strive to hire people with diverse backgrounds and experience. Our first product, Pi, provides an empathetic and conversational chatbot. Pi is a first public instance of building from our 350B+ frontier model with our sophisticated fine-tuning (10M+ examples), inference, and orchestration platform. We are now focusing on building new systems that directly support the needs of enterprise customers using this same approach Want to work with us? Have questions? Apply to job below or reach out to us with questions!