Taro Logo

Post-training - Model Fusion Research Engineer

AI research and deployment company dedicated to ensuring general-purpose artificial intelligence benefits all of humanity.
$360,000 - $440,000
Machine Learning
Staff Software Engineer
Hybrid
AI
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Post-training - Model Fusion Research Engineer

Our team is responsible for the post-training phase of ChatGPT, transforming large pre-trained models into powerful, safe, and user-friendly chatbots. We collaborate across the company to enhance ChatGPT's safety, speed, intelligence, utility, and overall capabilities. We integrate these improvements into the final models powering our production ChatGPT and API services, impacting millions of users worldwide.

We are seeking an engineer to accelerate the deployment of improvements to our models. You will collaborate with diverse teams handling various facets of the system, including core capabilities, multimodal integration (speech, images, and video), tools, and more. This role offers a unique opportunity to shape the future of ChatGPT, working across the technology stack, from optimizing low-level components like GPU kernels and network traffic to mastering the intricacies of RL post-training.

The ideal candidate has a robust technical background in areas such as data technologies, reliable software engineering, production ML model development, and cross-functional collaboration. While research experience is not required, a deep understanding of ML fundamentals and large-scale deep learning is essential for troubleshooting and analyzing complex system and ML issues. Excellent verbal and written communication skills, along with strong project management abilities, are crucial as you will collaborate with both technical research teams and non-technical product teams across the company.

Key Responsibilities:

  • Lead the regular release process of ChatGPT and API models
  • Dive deep into large ML codebases to diagnose and resolve systems and ML issues
  • Ensure a stable, smooth, and user-friendly model development process
  • Collaborate with cross-functional teams to design new features and integrate enhancements into ChatGPT

Sample projects include:

  • Consulting with product teams on how to best improve ChatGPT
  • Integrating new multimodal features or tools into ChatGPT
  • Working with data scientists to understand the impact of new models on users

This role is based in San Francisco, CA, with a hybrid work model of 3 days in the office per week. Relocation assistance is offered to new employees.

Last updated 8 months ago

Responsibilities For Post-training - Model Fusion Research Engineer

  • Lead the regular release process of ChatGPT and API models
  • Dive deep into large ML codebases to diagnose and resolve systems and ML issues
  • Ensure a stable, smooth, and user-friendly model development process
  • Collaborate with cross-functional teams to design new features and integrate enhancements into ChatGPT
  • Consult with product teams on how to best improve ChatGPT
  • Integrate new multimodal features or tools into ChatGPT
  • Work with data scientists to understand the impact of new models on users

Requirements For Post-training - Model Fusion Research Engineer

Python
Kubernetes
  • Robust technical background in data technologies, reliable software engineering, and production ML model development
  • Deep understanding of ML fundamentals and large-scale deep learning
  • Excellent verbal and written communication skills
  • Strong project management abilities
  • Experience working in complex technical environments
  • Experience debugging ML systems
  • Experience with reinforcement learning and/or transformers
  • Experience with Python
  • Experience with Kubernetes / distributed infrastructure
  • Experience with GPUs
  • Experience with one or more large scale data systems such as Beam or Spark

Benefits For Post-training - Model Fusion Research Engineer

Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Assistance
401k
Parental Leave
Education Budget
Equity
Relocation Benefits
  • Medical, dental, and vision insurance for you and your family
  • Mental health and wellness support
  • 401(k) plan with 50% matching
  • Unlimited time off and 13 company holidays per year
  • Paid parental leave (24 weeks) and family-planning support
  • Annual learning & development stipend ($1,500 per year)
  • Generous equity
  • Relocation assistance

Interested in this job?