Note: The job is a remote job and is open to candidates in USA. Talkspace is the world’s leading online therapy company, dedicated to transforming mental healthcare access. They are looking for a Senior AI Engineer to design, build, and scale autonomous AI agents for their behavioral health platform, focusing on safety, ethical autonomy, and data privacy.
Responsibilities
- Design and implement the technical architecture for Tee's core AI agents, including the development of planning modules, memory/retrieval systems, goal-setting algorithms, and tool-use orchestration
- Apply advanced RL, Inverse RL, or related control theory methods to develop agents capable of adaptive, long-term intervention strategies that maximize positive user outcomes while minimizing risk (e.g., optimizing interaction sequencing, timing, and content)
- Select, fine-tune, and deploy foundation models (LLMs) to power agent reasoning, natural language understanding, and empathetic, context-aware communication with users
- Develop models for human-agent interaction (HAI), incorporating principles from cognitive science and behavioral economics to ensure agents are effective, trustworthy, and aligned with therapeutic protocols
- Construct robust simulation environments for pre-training and testing agent policies, ensuring system stability and safety across a wide range of psychological and behavioral scenarios
- Optimize the deployment environment to manage the computational demands of multi-agent orchestration, ensuring low-latency decision-making and high system resilience
- Lead the evaluation and adoption of new agentic frameworks, reasoning technologies, and system design patterns that position Tee as a leader in autonomous behavioral health technology
- Design and implement comprehensive evaluation pipelines for multi-agent orchestration—including visualisations, trace-level analysis of LLM calls and tool invocations, offline evaluation against golden datasets, real-time production monitoring for behavioral drift and outcome correlation, guardrails, and human-in-the-loop annotation workflows. Establish metrics frameworks to assess reasoning quality, task completion, safety compliance, and task alignment across the full agent lifecycle
Skills
- Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Robotics, Electrical Engineering, or a highly quantitative field
- Minimum of 5+ years of experience in a production software engineering environment, with at least 3+ years specifically focused on designing, implementing, and deploying complex machine learning or autonomous systems
- Exceptional Python Programming Skills: Mastery of Python and its scientific libraries
- Production MLOps Expertise: Demonstrated experience building and managing CI/CD pipelines for ML models and integrating them into service-oriented architectures (SOA)
- Autonomous/Agentic Systems: Strong theoretical understanding and practical application of techniques central to agentic AI, such as Reinforcement Learning (RL), planning algorithms, or complex decision-making systems
- LLM Deployment: Hands-on experience fine-tuning, deploying, and managing Large Language Models (LLMs) in a production environment, including knowledge of prompt engineering, retrieval-augmented generation (RAG), and cost optimization
- Software Engineering Rigor: Proven ability to write high-quality, maintainable, scalable, and well-tested production code
- Proficiency with cloud infrastructure (AWS, GCP) and containerization technologies (Docker, Kubernetes)
Benefits
- Comprehensive Medical, Dental and Vision plans coverage since day one
- Pre-tax benefits: HSA/ FSA
- 401k Retirement Savings Program with matching up to 4%
- Voluntary benefits including disability, basic life or pet insurance, etc.
- Monthly Wellness Stipend to promote mental and physical self-care
- Flexible PTO and Remote First Environment
- Regular team events, including Wellness Workshops and Team Building Events
- Free access to Talkspace products for you and one household member, as well as access to a friends and family discount!
Company Overview