Staff AI Engineer

This is a fully remote position, open to applicants in United States.

📋 Description

• Take charge of the intelligence layer within the platform, encompassing LLM orchestration, retrieval pipelines, and evaluation frameworks.

• Enhance LangGraph orchestration for intricate clinical workflows.

• Develop retrieval pipelines to query patient records utilizing embeddings and vector similarity.

• Implement Langfuse/LangSmith tracing to troubleshoot hallucinations in summarization chains.

• Fine-tune models focusing on medical terminology or the structure of clinical notes.

• Assess Gemini, Claude, and GPT for tasks in the healthcare sector.

• Create dependable and ethical pipelines that convert unstructured clinical documents into searchable vectors.

• Write TypeScript and Python services that link the AI systems together.

• Deploy work to production within a small, remote-first, and asynchronous-friendly team.

⛳️ Requirements

• Practical experience with LLMs using OpenAI, Anthropic, or Google APIs.

• Knowledge of context windows and temperature settings.

• Familiarity with LangChain, LangGraph, or comparable orchestration tools.

• Experience in chaining calls, managing failures, and overseeing state.

• Proficiency with Langfuse, LangSmith, or Phoenix.

• Strong background with agentic code authoring systems such as Claude, Codex, Gemini, Copilot, or similar technologies.

• Retrieval/RAG experience with Pinecone, Weaviate, pgvector, or Vertex AI Matching Engine.

• Understanding of chunking strategies, embedding models, and reranking techniques.

• Capability to write Python for ML pipelines and TypeScript for services.

• Experience with fine-tuning using LoRA, PEFT, or complete fine-tunes (bonus).

• Background in healthcare/FHIR/clinical data (bonus).

• Familiarity with Gemini models and Vertex AI (bonus).

• Knowledge of evaluation frameworks such as RAGAS, custom evaluations, or human-in-the-loop setups (bonus).

• Experience in making embeddings effective on messy, real-world data (bonus).

• Must already be authorized to work in the United States without employer sponsorship.

🏝️ Benefits

• Competitive salary plus equity.

• Health insurance coverage.

• Remote-first and asynchronous-friendly work environment.

• Small team dynamic where your contributions directly lead to production delivery, rather than being stuck in a backlog.