Member of Technical Staff, Essential AI

Research Engineer, Pre-Training

Salary not provided

Python

PyTorch

Junior, Mid and Senior level

San Francisco Bay Area

More information about location

AI products for enterprise

Open for applications

AI products for enterprise

1-20 employees

B2BArtificial IntelligenceEnterpriseInternal toolsBusiness IntelligenceSaaS

Open for applications

Salary not provided

Python

PyTorch

Junior, Mid and Senior level

San Francisco Bay Area

More information about location

1-20 employees

B2BArtificial IntelligenceEnterpriseInternal toolsBusiness IntelligenceSaaS

Company mission

To deepen the partnership between humans and computers, unlocking collaborative capabilities that far exceed what could be achieved today.

Job

Company

Role

Who you are

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

What the job involves

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
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

Our take

Essential AI shifted out of stealth mode in 2023 with the $56.5 million backing of Google and Nvidia, no mean feat for a company founded the year before. It develops full-stack AI products to support productivity by automating time-consuming and monotonous workflows.

Bringing forward experience and knowledge from careers in giants like Google, the company is fuelled with severe understanding of the sector. Though in its primitive stages, its promise of benefiting enterprise operations by deepening the relationship between computers and humans led to its exciting raise in funding.

This Series B will allow the company to build a diverse team to delve into a new market, building tools that will drastically change the scale productivity for its customers. With ambitious goals, we’re excited to see the specifics of what Essential AI will bring to the space moving forward.

Freddie

Company Specialist at Welcome to the Jungle

Company

Funding (2 rounds)

Dec 2023

$56.5m

SERIES A

Mar 2023

$8.3m

SEED

Total funding: $64.8m

Company HQ

SoMa, San Francisco, CA

Leadership

Ashish Vaswani

(Co-founder & CEO)

Previously Co-founder and Chief Scientist of Adept AI Labs and Staff Research Scientist at Google Brain.

Niki Parmar

(Co-founder)

Was Co-founder and CTO at Adept AI Labs. Worked at Google for over 6 years through several roles including Staff Research Scientist for Google Brain.

Salary benchmarks

We don't have enough data yet to provide salary benchmarks for this role.

Submit your salary to help other candidates with crowdsourced salary estimates.

Share this job

View 1 more job at Essential AI