Member of Technical Staff: Research Engineer, Pre-Training

4 weeks ago


San Francisco, United States Essential AI Full time

Essential AI's mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user value per FLOP.

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. We are well-capitalized and supported by March Capital and Thrive Capital, with participation from AMD, Franklin Venture Partners, Google, KB Investment, NVIDIA.
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. We offer relocation assistance to new employees.

  • San Francisco, United States Essential AI Full time

    Essential AI's mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user value per FLOP. We believe that a...


  • San Francisco, United States essential AI Full time

    Job DescriptionJob DescriptionEssential AI’s mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user...


  • San Francisco, United States essential AI Full time

    Job DescriptionJob DescriptionEssential AI’s mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user...


  • San Francisco, United States Essential AI Full time

    Essential AI's mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user value per FLOP. We believe that...


  • San Francisco, United States essential AI Full time

    Essential AI's mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user value per FLOP. We believe that...


  • San Francisco, United States essential AI Full time

    Essential AI's mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user value per FLOP. We believe that...


  • San Francisco, United States Essential AI Full time

    Essential AI's mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user value per FLOP. We believe that...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology: Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world. The Role: As a Member of the Technical Staff, you will join a team with deep...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology: Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world. The Role: As a Member of the Technical Staff, you will join a team with deep...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology: Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world. The Role: As a Member of the Technical Staff, you will join a team with deep...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology: Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world. The Role: As a Member of the Technical Staff, you will join a team with...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology: Check you match the skill requirements for this role, as well as associated experience, then apply with your CV below. Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology:Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world.The Role:As a Member of the Technical Staff, you will join a team with deep...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology:Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world.The Role:As a Member of the Technical Staff, you will join a team with deep...


  • San Francisco, United States Kiso Technology Full time

    Kiso Technology:Kiso Technology is a boutique technical consultancy based in San Francisco. We develop algorithms, models, and software systems to solve complex problems in a variety of domains. Clients range from early-stage startups to some of the largest companies in the world.The Role:As a Member of the Technical Staff, you will join a team with deep...


  • San Francisco, United States Acceler8 Talent Full time

    Member of Technical Staff: Machine Learning Introduction As a Member of Technical Staff specializing in Machine Learning, you have the opportunity to join a pioneering team dedicated to enhancing the interplay between humans and AI technology. Our vision focuses on integrating cutting-edge ML techniques to optimize user interactions across various platforms,...


  • San Francisco, United States Acceler8 Talent Full time

    Member of Technical Staff: Machine Learning Like the look of this opportunity Make sure to apply fast, as a high volume of applications is expected Scroll down to read the complete job description. Introduction As a Member of Technical Staff specializing in Machine Learning, you have the opportunity to join a pioneering team dedicated to enhancing the...


  • San Francisco, United States Acceler8 Talent Full time

    Member of Technical Staff: Machine LearningIntroduction As a Member of Technical Staff specializing in Machine Learning, you have the opportunity to join a pioneering team dedicated to enhancing the interplay between humans and AI technology. Our vision focuses on integrating cutting-edge ML techniques to optimize user interactions across various platforms,...


  • San Francisco, United States Acceler8 Talent Full time

    Member of Technical Staff: Machine LearningIntroduction As a Member of Technical Staff specializing in Machine Learning, you have the opportunity to join a pioneering team dedicated to enhancing the interplay between humans and AI technology. Our vision focuses on integrating cutting-edge ML techniques to optimize user interactions across various platforms,...


  • San Francisco, United States essential AI Full time

    Job DescriptionJob DescriptionEssential AI’s mission is to deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today. We believe that building delightful end-user experiences requires innovating across the stack - from the UX all the way down to models that achieve the best user...