ML Infra Engineer

Immunai; United States; Hybrid; USD 88,000 - 122,000; Full time

Job summary

Immunai in Ramat Gan, Israel, is seeking an ML Infra Engineer to own benchmarking and evaluation of foundation models and multimodal systems. You will support model development, iteration, and validation, and build out the ML engineering infrastructure for evaluation and deployment.

This role sits at the intersection of software engineering and applied AI, shaping how models are built and compared across the organization, with a hybrid work model.

Qualifications

• Bachelor's/Master's/PhD in computer science or related field.
• Strong software engineering skills with modular systems.
• Hands-on experience with ML models and evaluation pipelines.
• Proficiency in Python and modern ML ecosystems.
• Ability to read, modify, and debug deep learning models.
• Experience with benchmarks, metrics, or evaluation frameworks (preferred).
• Familiarity with foundation models or multimodal learning (preferred).
• Comfort navigating complex datasets and doing targeted exploratory analysis.
• Experience in biomedical or data-intensive domains (a plus).

Responsibilities

• Own & evolve benchmarking for foundation models and multimodal AI systems.
• Define core abstractions for datasets, tasks, models, metrics, and evaluation workflows.
• Develop metrics capturing predictive performance, biological relevance, and multimodal alignment.
• Support model development by collaborating with AI scientists and data scientists.
• Bring in new models and baselines to benchmarks and ensure meaningful comparisons.
• Explore data and results to debug evaluations and understand model behavior.
• Ensure evaluations are versioned, consistent, and reproducible over time.

Skills

Python

ML evaluation

Software engineering

Data exploration

Education

BSc/MSc/PhD in CS or related

Job description

About Immunai:

Immunai is an AI-driven platform company focused on improving drug discovery and development by decoding the human immune system.

We combine large-scale single cell immune data, advanced machine learning, and strong engineering to help pharmaceutical and research partners make better, more informed decisions throughout the drug development process.

Our long-term goal is to reduce drug development failure rates and help more effective medicines reach patients. We're building this platform thoughtfully and collaboratively, bringing together expertise across biology, AI, engineering, and business.

Immunai is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees.

About the role:

We are looking for an ML Infra Engineer to play a central role in evaluating, and deploying Immunai's AI models. You will own and evolve Immunai's benchmarking and evaluation capabilities for foundation models and multimodal systems, while also working closely with modeling teams to support model development, iteration, and validation. This role sits at the intersection of software engineering, model understanding, and applied AI, with broad influence on how models are built, compared, and improved across the organization.

This is NOT a core algorithmic research role - but rather, a role ++focused on building out the ML engineering infrastructure++ for the evaluation and deployment of models.

Location:

Ramat Gan, Israel (hybrid model)

What will you do?

• Own & Evolve Benchmarking -- Design, build, and maintain Immunai's benchmarking suite for foundation models and multimodal AI systems.
• Define Core Abstractions -- Create clean, extensible abstractions and APIs for datasets, tasks, models, metrics, and evaluation workflows.
• Develop Metrics & Evaluations -- Implement metrics that capture predictive performance, biological relevance, and multimodal alignment.
• Support Model Development -- Work closely with AI scientists and data scientists to integrate new models, tweak architectures, and enable rapid, fair iteration.
• Bring in New Models & Baselines -- Add external and internal models to benchmarks and ensure meaningful comparisons.
• Explore Data When Needed -- Dive into data and results to debug evaluations, understand model behavior, and unblock modeling work.
• Enable Rigor & Reproducibility -- Ensure evaluations are consistent, well‑versioned, and trustworthy over time.

Required qualifications:

• BSc, MSc, or PhD in Computer Science, Software Engineering, or a related field
• Strong software engineering skills with experience designing maintainable, modular systems
• Hands‑on experience working with ML models and evaluation pipelines
• Proficiency in Python and modern ML ecosystems
• Ability to read, modify, and debug deep learning models
• Experience with benchmarks, metrics, or evaluation frameworks --- preferred
• Familiarity with foundation models or multimodal learning --- preferred
• Comfort navigating complex datasets and doing targeted exploratory analysis
• Experience in biomedical or other data‑intensive domains --- a plus

Desired personal traits:

• You want to make an impact on humankind
• You prioritize "We" over "I"
• You enjoy getting things done and striving for excellence
• You collaborate effectively with people of diverse backgrounds and cultures
• You have a growth mindset
• You are candid, authentic, and transparent

When you apply for a role at Immunai, we process your personal data (such as CV, contact details, and professional background) to assess your suitability for current and future roles. Your information may be shared internally with relevant employees and stored or processed by trusted service providers, some of whom may be located outside your country of residence. You can read more about how we process data here: https://www.immunai.com/privacy-policy/.