Staff Software Engineer, AI/ML

D24 Search is the world's largest virtual clinic for women and families, and they are seeking a Staff Software Engineer specializing in AI/ML to join their foundational AI team. The role involves designing, building, and deploying AI-powered solutions that directly impact millions of users, while working with cutting-edge technologies and shaping the AI strategy as the organization scales.

Responsibilities:

• Design, build, and deploy end-to-end AI-powered solutions using models like OpenAI and modern orchestration techniques (RAG, agents)
• Build critical AI initiatives such as our chatbot, recommendation engine, and search functionalities
• Develop robust ML pipelines and platform components for model experimentation, fine-tuning, and deployment with a focus on scalability, safety, and performance
• Collaborate cross-functionally with product, data, and engineering teams to turn rich healthcare data into actionable insights and intelligent workflows
• Drive architecture discussions and technical decision-making across AI/ML systems, promoting best practices

Requirements:

• 6 - 15 years of experience in a software engineering role
• Strong proficiency in Python for software and ML development
• Experience with LLMs, RAG, and agent frameworks in production
• Experience with PyTorch or TensorFlow
• Experience as a full stack SWE with a strong back-end lean OR ML engineers with experience building AI/ML products end-to-end
• Experience with Generative AI applications
• Experience working end-to-end on the ML pipeline (e.g., data ingestion and structuring through analysis and training etc.)
• Experience shipping a product or feature from 0 to 1
• Bachelor's degree in CS, or Math, Physics, or related engineering field
• Possesses a builder mindset: Excited to be part of a foundational team, adaptable, and actively learning as the AI landscape changes
• Track record of agency and autonomy: comfortable being accountable for delivery without heavy mentorship/support
• Must be located in San Francisco, Seattle, or New York City