Skip to content
CollateCollate

AI Engineer

Builds and productionizes AI models and systems for life sciences document generation, focusing on LLMs, NLP robustness, and reliable deployment pipelines. Bridges ML research, software engineering, and product needs in a high-stakes domain.

About the job

What You’ll Do

  • Prompt optimization, red-teaming, and improving robustness of LLMs for life-science NLP applications within Collate’s products.
  • Build pipelines and infrastructure to deploy AI systems reliably, safely, and at scale.
  • Collaborate with product, design, and engineering teams to translate user needs into AI-driven features.
  • Develop evaluation frameworks to ensure models are accurate, fair, and trustworthy in real-world life science settings.
  • Experiment rapidly, while balancing iteration speed with the rigor required for high-stakes applications.
  • Create tools and workflows that make AI development more efficient across the team.

What We’re Looking For

  • Hands-on experience building and deploying ML/AI systems in production.
  • Strong foundation in deploying search and retrieval systems and NLP applications.
  • Ability to bridge research and engineering — from prototyping models to shipping them in user-facing products.
  • Comfort working in an early-stage startup where ambiguity is high and ownership is expected.
  • Motivation to apply AI to life sciences in a way that prioritizes reliability, safety, and impact.

Skills

LLMs, NLP, Machine Learning, Prompt Engineering, Red-Teaming, Retrieval Systems, Search Systems, MLOps, AI Infrastructure, Evaluation Frameworks

Thinking Machines Lab

Thinking Machines Lab

San Francisco, CA

Research Software Engineer, Post Training
$350k+/yrHybridML Engineering

Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.

OpenAI

OpenAI

San Francisco, CA

Software Engineer, AI for Chip Design
$266k+/yrHybridML Engineering

Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.

Rollstack

Rollstack

United States
AI Software Engineer
No salary listedRemote3+ YOEML Engineering

Build production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.

ClickUp

ClickUp

United States

Machine Learning Engineer, Ranking & Retrieval
$200k+/yrRemote5+ YOEML Engineering

Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.

PathAI

PathAI

Boston, MA
Machine Learning Engineer III
$131k+/yrOn-site5+ YOEML Engineering

Develop and deploy machine learning models for biomedical research and AI products, collaborating with scientific, engineering, and product teams. Requires an advanced quantitative degree, substantial ML experience, Python proficiency, and experience bringing models into production or research applications.