Machine Learning Research Engineer, Scientific AI

Just Food Company; Emeryville (CA); On-site; USD 160,000 - 200,000; Full time

Benefits offered by this job

Equity

Up to 100% employer-paid medical, dent

Flexible time off

Job summary

Just Food Company seeks a Machine Learning Research Engineer to advance protein discovery and AI-enabled products. You will build ML models that connect protein sequence, structure, and experimental data to functional food performance, working with protein scientists, food scientists, and product developers.

This highly hands-on role blends research with practical deployment across the company, offering ownership to design experiments, build tooling, and scale AI solutions in collaboration with

Qualifications

• BS, MS, or PhD in Computer Science, Machine Learning, Computational Biology, or a related quantitative field.
• Open to candidates from industry, academia, or open-source backgrounds.
• Strong ability to translate research ideas into working ML systems.
• Experience with research-to-product handoff and reproducible workflows.

Responsibilities

• Develop models predicting food functionality from protein sequence, structure, and experimental data.
• Build representations linking proteins, ingredients, and formulations to outcomes.
• Explore protein representation learning, graph/deep learning, generative models, and Bayesian optimization.
• Collaborate with experimental scientists to design studies and incorporate results.
• Create data pipelines, training infra, and evaluation frameworks for iterative modeling.
• Translate research into practical tools used by scientists and product teams.

Skills

ML modeling

Python

PyTorch

JAX

TensorFlow

Model deployment

Data analysis

Communication

Education

BS/MS/PhD in CS/ML/Quant field

Tools

PyTorch

JAX

TensorFlow

GPU/Distributed training

Job description

Machine Learning Research Engineer, Scientific AI

Just Food Company is on a mission to build a healthier, safer, and more sustainable food system—starting with what’s on your plate. From pioneering plant-based eggs to making real meat without slaughter, we’re reimagining the future of food in ways that are better for people and the planet.

We’re a lean, passionate team of scientists, chefs, engineers, and operators who believe that doing hard things well can make a meaningful difference. We’ve brought groundbreaking innovations to market, like Just Egg and GOOD Meat, and we’re just getting started.

If you’re excited by big challenges, real impact, and the chance to shape what the future tastes like—we’d love to meet you.

Just Food Company uses science and technology to build a better food system.

Our work brings together protein science, food science, machine learning, product development, and commercial execution. We investigate proteins, understand how their properties translate into food functionality, and use those insights to create products that can compete on taste, performance, nutrition, affordability, and scale.

We believe machine learning can fundamentally change both how we conduct scientific research and how we operate as a company. We are looking for an exceptional Machine Learning Research Engineer to help us build that capability.

This role has two connected mandates.

The primary focus is R&D. You will build machine learning systems that help us identify, evaluate, and validate proteins with valuable food functionalities. Our goal is to create a faster learning loop between prediction and experimentation: identify promising proteins, predict how they may perform in food, select the most informative experiments, and use experimental results to continuously improve subsequent predictions and decisions.

The second mandate is applying AI across Just Food Company. You will work with teams across the organization to identify important workflows that AI can materially improve and build practical systems that help people access knowledge, analyze information, make decisions, and execute their work more effectively.

This is an opportunity for someone who wants to apply modern machine learning to challenging scientific problems while also helping shape how an organization uses AI more broadly.

About the Role

As a Machine Learning Research Engineer, you will work at the intersection of machine learning, scientific research, and practical implementation.

Within R&D, you will develop models that connect protein sequence, structure, biochemical properties, and experimental data to functional performance in food. You will work closely with protein scientists, food scientists, and product developers to determine which proteins to investigate, what experiments to run, and how to learn systematically from every result.

Depending on the problem, your work may involve protein representation learning, graph and geometric deep learning, multimodal models, generative or diffusion models, active learning, Bayesian optimization, or other approaches. We are not committed to a particular model family. We care about selecting the right method for the scientific problem and rigorously evaluating it against real experimental outcomes.

Beyond R&D, you will apply the same problem-solving mindset to opportunities across the company. Working directly with teams in product development, sales, operations, and other functions, you will identify high-value applications for AI, rapidly develop and evaluate solutions, and turn successful prototypes into reliable tools.

This is a highly hands‑on role. You will have significant ownership and autonomy while working closely with scientists, functional teams, and company leadership.

What You’ll Do

Accelerate Scientific Discovery

• Develop models that predict food functionality using protein sequence, structure, biochemical/biomaterial properties, formulations including other proteins/ingredients, and experimental data.
• Build representations and models that capture relationships among proteins, ingredients, formulations, processing conditions, and functional outcomes.
• Explore approaches such as protein representation learning, graph and geometric deep learning, generative and diffusion models, multimodal learning, active learning, and Bayesian optimization where appropriate.
• Partner directly with experimental scientists to design studies, evaluate predictions, and incorporate experimental results into subsequent modeling.
• Build data pipelines, training infrastructure, evaluation frameworks, and research tools to support iterative model development.
• Develop prospective evaluations to determine whether model predictions improve protein selection and experimental decision‑making.
• Translate successful research into practical tools that scientists and product developers can use.

Apply AI Across Just Food Company

• Partner with product development, sales, operations, and other teams to identify high‑value opportunities for AI.
• Build practical systems that improve knowledge access, analysis, planning, reporting, and execution.
• Develop applications and agents that can perform useful work using our data, knowledge, and systems.
• Rapidly prototype potential solutions, evaluate their value, and turn the strongest ideas into reliable tools.
• Apply appropriate permissions, testing, human oversight, and measurement to deployed systems.
• Help establish shared infrastructure and development practices that allow AI systems to be built and used reliably across the company.
• Measure success through adoption, time saved, decision quality, scientific impact, and work completed — not simply the number of tools launched.

What We’re Looking For

• Strong foundations in machine learning and demonstrated experience developing modern machine learning models and systems.
• Excellent Python skills and hands‑on experience with PyTorch, JAX, TensorFlow, or a comparable machine learning framework.
• Experience taking machine learning problems from data and experimentation through evaluation and implementation.
• Ability to reason across data, model architecture, training, evaluation, inference, and deployment.
• Ability to select methods based on the structure of a problem rather than defaulting to a particular model or technology.
• Comfort working with sparse, noisy, heterogeneous, or partially observed data.
• Strong software engineering fundamentals and a commitment to testing, reproducibility, maintainability, and performance.
• Ability to balance open‑ended research with practical implementation and iteration.
• Intellectual curiosity and the ability to quickly develop fluency in unfamiliar scientific and business domains.
• Strong communication skills and enthusiasm for working directly with scientists, product developers, operators, commercial teams, and company leadership.
• Comfort operating with significant ownership and autonomy in a small, fast‑moving organization.

Particularly Relevant Experience

You do not need experience in all of the areas below. Strong candidates may bring depth in one or more of the following:

• Protein representation learning, protein language models, graph or geometric deep learning, diffusion or other generative models, multimodal learning, active learning, or Bayesian optimization.
• Biological sequences, protein structures, molecular graphs, biochemical measurements, or other structured scientific data.
• Computational biology, bioinformatics, protein engineering, structural biology, chemistry, food science, ingredient discovery, or another scientific ML domain.
• Designing computational approaches alongside laboratory scientists and incorporating experimental results into subsequent modeling.
• Language‑model applications, retrieval systems, tool‑using agents, or internal AI applications.
• Data pipelines, GPU workloads, cloud infrastructure, distributed training, or scientific computing.
• Research, open‑source projects, publications, patents, datasets, or production machine learning systems that demonstrate exceptional technical ability.

Education & Experience

Candidates may come from a range of educational and professional backgrounds.

A BS, MS, or PhD in Computer Science, Machine Learning, Computational Biology, Bioinformatics, Applied Mathematics, Statistics, or a related quantitative field is relevant, but we value demonstrated ability over a particular degree or academic pedigree.

We are open to candidates at different stages of their careers. You may have developed your expertise through industry, academic research, open‑source work, independent projects, or a combination of these experiences.

Previous food science experience is not required, nor is experience with every technology or model family listed in this posting.

What matters most is exceptional machine learning ability, strong engineering fundamentals, scientific curiosity, and evidence that you can learn quickly and turn difficult problems into working solutions.

What Success Looks Like

Within R&D, success means helping us identify better protein candidates, choose more informative experiments, predict functional outcomes more effectively, and learn faster from experimental results.

Across Just Food Company, success means identifying high‑value opportunities for AI and building practical systems that measurably improve how teams access information, make decisions, and execute their work.

Ultimately, success means creating a tighter loop between data, prediction, experimentation, learning, and action — both in our science and across the company.

The total package

We develop our salaries using market data, internal benchmarks, and candidate experience to ensure fairness and competitiveness. All full‑time team members receive:

• Equity
• Up to 100% employer‑paid medical, dental, and vision benefits (up to90% for dependents)
• Flexible time off

Compensation: The expected base salary range for this role (based inEmeryville, CA/full‑time) is $160,000 – $200,000 per year. Final compensation will be based on skills, experience, and overall contribution to the role.

Additional Information & Requirements

This is a full‑time regular position basedonsite(5 days/week)at our headquarters inEmeryville, CA.We are prioritizing local candidatesat this time.

At this time, Just Food Company is unable to sponsor or transfer employment visas.Candidates mustbe authorized towork in the United States without current or future employer sponsorship.

In compliance with federal law, all new hires must verify their identity and eligibility to work in the United States and complete the required Form I-9 upon hire.

Just Food Companyand its subsidiariesparticipatein E-Verify.