Demo

Member of Technical Staff: Research Engineer, Pre-Training

Essential AI
San Francisco, CA Full Time
POSTED ON 2/19/2025
AVAILABLE BEFORE 5/17/2025

About Us

We believe that a small, focused team of motivated individuals can create outsized breakthroughs. We are building a world-class multi-disciplinary team who are excited to solve hard real-world AI problems.

Working backward from the mission - "Mimic and amplify our greatest strength - the ability to solve unseen problems", we aim to cover the pieces that can build on the essential path towards building SOTA capabilities, tooling, and talent.

If you don't see an open role but still greatly want to contribute towards building next-generation AI research and development, please reach out to hiring@essential.ai

The Role

The Research Engineer, Pre-Training will be responsible for designing and implementing novel pre-training approaches to create powerful foundation models that can be fine-tuned / further aligned for a variety of downstream tasks. You will work very closely with our Research Scientists to identify key challenges and opportunities, and then develop and test new pre-training techniques and architectures. This may involve exploring different model architectures, training objectives, data sources, and scaling approaches. You will also be responsible for running large-scale experiments, analyzing results, and iterating on your approaches.

What you'll be working on

  • You will lead or be a core contributor to our research bets that advance the the real-world capabilities of our models.
  • You will collaborate closely with our data and product teams to close the loop between research and product, identify capability gaps and evaluate progress.
  • Design novel pre-training architectures and algorithms to improve model performance and efficiency.
  • Design and execute experiments to evaluate the efficacy of pre-training techniques across various datasets; Analyze experimental results to gain insights into model behavior and identify areas for improvement.
  • Develop efficient and scalable pre-training pipelines to train models on massive amounts of data
  • Implement pre-training models and algorithms; Optimize model performance and scalability for deployment in production environments.

What we are looking for

  • Self-motivated and takes a proactive approach to constantly iterate by continuously experimenting, inferring, and deciding the right set of next experiments.
  • Research experience with a focus on pre-training and building large language models using frameworks such as Megatron, DeepSpeed, MaxText, etc.
  • You have strong ML fundamentals and first principles thinking that guides your approach to research.
  • You have experience of coming up with new methods or improving existing techniques in ML or related fields
  • Proficiency in programming languages such as Python and frameworks such as JAX, PyTorch or TF
  • Experience with data engineering and preprocessing, in particular, optimization of data pipelines, feature engineering, and model evaluation is beneficial.
  • Strong problem solving, analytical, communication, and collaboration skills with the ability to analyze complex datasets and derive actionable insights.
  • Ability to prototype and deploy pre-trained models in production environments
  • You enjoy building things from the ground up in a fast-paced, collaborative environment.
  • We encourage you to apply for this position even if you don't check all of the above requirements but want to spend time pushing on these techniques.

    We are based in-person in SF and work fully onsite 5 days a week. We offer relocation assistance to new employees.

    The base pay range target for the role seniority described in this job description is up to $225,000 in San Francisco, CA. Final offer amounts depend on various job-related factors, including where you place on our internal performance ladders, which is based on factors including past work experience, relevant education, and performance on our interviews and our benchmarks against market compensation data. In addition to cash pay, full-time regular positions are eligible for equity, 401(k), health benefits, and other benefits like daily onsite lunches and snacks; some of these benefits may be available for part-time or temporary positions.

    Essential AI commits to providing a work environment free of discrimination and harassment, as well as equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. You may view all of Essential AI's recruiting notices here, including our EEO policy, recruitment scam notice, and recruitment agency policy.

    Salary : $225,000

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Member of Technical Staff: Research Engineer, Pre-Training?

    Sign up to receive alerts about other jobs on the Member of Technical Staff: Research Engineer, Pre-Training career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $84,355 - $134,776
    Income Estimation: 
    $135,949 - $221,193
    Income Estimation: 
    $135,356 - $164,911
    Income Estimation: 
    $153,053 - $187,211
    Income Estimation: 
    $153,902 - $198,246
    Income Estimation: 
    $113,077 - $147,784
    Income Estimation: 
    $135,356 - $164,911
    Income Estimation: 
    $153,902 - $198,246
    Income Estimation: 
    $98,763 - $126,233
    Income Estimation: 
    $116,330 - $143,011
    Income Estimation: 
    $113,077 - $147,784
    Income Estimation: 
    $116,330 - $143,011
    Income Estimation: 
    $135,356 - $164,911
    Income Estimation: 
    $153,902 - $198,246
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Not the job you're looking for? Here are some other Member of Technical Staff: Research Engineer, Pre-Training jobs in the San Francisco, CA area that may be a better fit.

    Member of Technical Staff - AI Research Engineer

    Future House USA, San Francisco, CA

    AI Assistant is available now!

    Feel free to start your new journey!