Machine Learning Research Engineer, Scientific AI

The role

You will build machine learning systems to identify and validate proteins for food functionality. You will also apply AI across the organization to improve workflows and decision-making.

What you will do

• Develop models that predict food functionality using protein sequence, structure, and experimental data.
• Build representations and models capturing relationships among proteins, ingredients, and processing conditions.
• Partner with experimental scientists to design studies and incorporate results into modeling.
• Build data pipelines, training infrastructure, and evaluation frameworks.
• Translate research into practical tools for scientists and product developers.
• Identify high-value AI opportunities across product development, sales, and operations.
• Build practical systems to improve knowledge access, analysis, and execution.
• Develop applications and agents to perform work using company data and systems.
• Rapidly prototype and evaluate potential AI solutions.
• Help establish shared infrastructure and development practices for AI.

Required qualifications

• Strong foundations in machine learning.
• Demonstrated experience developing modern machine learning models and systems.
• Excellent Python skills.
• 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 problem structure.
• Comfort working with sparse, noisy, heterogeneous, or partially observed data.
• Strong software engineering fundamentals.
• Commitment to testing, reproducibility, maintainability, and performance.
• Ability to balance open-ended research with practical implementation.
• Authorized to work in the United States without current or future employer sponsorship.

Nice to have

• 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, or ingredient discovery.
• Designing computational approaches alongside laboratory scientists.
• 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.

Benefits

• Competitive base salary
• Equity
• Up to 100% employer-paid medical, dental, and vision benefits (up to 90% for dependents)
• Flexible time off

Before you apply

• 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.
• New hires must verify identity and eligibility to work in the United States and complete Form I-9.
• Just Food Company and its subsidiaries participate in E-Verify.

About the company

Just Food Company uses science and technology to build a better food system. They work with protein science, food science, and machine learning to create products that compete on taste, performance, nutrition, affordability, and scale.