Machine Learning Research Engineer

About Us

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: 

• Competitive base salary 

• Equity 

• Up to 100% employer-paid medical, dental, and vision benefits (up to 90% for dependents) 

• Flexible time off 

Compensation: The expected base salary range for this role (based in Emeryville, 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 based onsite (5 days/week) at our headquarters in Emeryville, CA. We are prioritizing local candidates at this time.  

At this time, Just Food Company is unable to sponsor or transfer employment visas. Candidates must be authorized to work 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 Company and its subsidiaries participate in E-Verify.