Senior AI Software Engineer (Agentic Systems & LLM Infrastructure)

Americas - Remote, US - Remote, Canada - Remote

⚠️ We'll shut down after Aug 1st - try foo🦍 for all jobs in tech ⚠️

Function Health

Discover, track, and own your health with Function Health. With 100+ lab tests and personalized action place to identify lifestyle changes to promote longevity. Test your levels today.

View all jobs at Function Health

Apply now Apply later

Company Overview

Function was founded with a singular focus: empower you to live 100 healthy years. We’re doing that by using the best available technology to make sure people don't suffer or die a preventable death. Function has been recognized as one of Fast Company’s Most Innovative Companies of 2024, and is venture-backed by Andreessen Horowitz (a16z). Hundreds of thousands of members have joined Function to take control of their health. We are growing our team and seeking out world-class talent that deeply believes in our mission to positively impact global health, has a relentless bias toward action and a growth mindset. Function fosters a collaborative and dynamic environment, where every day we are building the future.

Role Overview

As a Senior AI Software Engineer, you will own the design and development of end-to-end AI systems, spanning backend infrastructure, model orchestration, and production-grade deployment. You’ll work closely with product, data, and infrastructure teams to integrate LLMs, agentic frameworks (LangGraph, LangChain, AutoGen), computer vision models, and vector-based reasoning systems into consumer-facing and internal tools. This role is highly technical, hands-on, and requires both deep systems expertise and practical AI application skills. If you’re a high-ownership engineer excited to ship AI-powered products at scale, this is your opportunity to make a real impact.

Key Responsibilities


  • Design, build, and scale distributed backend systems that power LLMs, agents, and AI infrastructure.
  • Integrate AI/LLM models into product systems (e.g., LangGraph, Agentic APIs, vector stores, orchestration layers).
  • Develop high-reliability APIs and services for real-time AI inference, telemetry, usage metering, and monitoring.
  • Own backend architecture decisions across authentication, billing, deployments, and developer platforms.
  • Collaborate closely with product, infra, and research teams to translate cutting-edge AI into production systems.
  • Ensure production-readiness through observability, CI/CD pipelines, and SRE-style reliability engineering.
  • Drive continuous improvement, pushing for faster iteration, scalability, and throughput.
  • Mentor peers and lead by example with a hands-on, high-ownership, startup-style mindset.


Must-Have Qualifications


  • 5+ years of experience building production-grade backend systems in Python, Go, or similar.
  • Deep experience with distributed systems, microservices, and event-driven architecture.
  • Strong grasp of AI/ML infrastructure, model inference, model serving, and agentic frameworks.
  • PostgreSQL, Redis, or similar data store expertise, including performance tuning and scaling.
  • Familiarity with vector databases, embeddings, and LLM-specific storage/search systems.
  • Experience deploying in cloud-native environments (AWS, GCP) using Kubernetes, Terraform, Docker.
  • Proficiency with API design, backend frameworks (e.g., FastAPI, Flask), and load balancing.
  • Hands-on experience with telemetry, monitoring, logging, and production issue triage.
  • Understanding of CI/CD workflows, developer tooling, and test automation at scale.
  • Ability to work in fast-paced, early-stage environments, making tradeoffs and shipping quickly.
  • Strong communication skills: clear, concise, and collaborative—even when disagreeing.
  • Bias for action: you’re a scrappy problem solver who thrives in ambiguity.


Nice-to-Have Qualifications


  • Experience with LangChain, LangGraph, AutoGen, OpenAI APIs, or other agent-based orchestration frameworks.
  • Familiarity with computer vision models and multimodal AI systems.
  • Background in health tech, regulated data systems (HIPAA compliance), or medical data pipelines.
  • Experience building AI-powered consumer applications with personalization and real-time inference.
  • Exposure to reinforcement learning, retrieval-augmented generation (RAG), or long-term memory agents.
  • Motivated by a mission-driven product, with excitement for being part of explosive growth.
  • Background in competitive sports, military, or other high-intensity, high-throughput environments where resilience and ownership are key.

Why Join Us


  • Work on cutting-edge AI products that directly help people improve their health.
  • Build agentic systems from the ground up, influencing technical strategy and architecture.
  • Join a fast-moving, high-ownership team, with significant autonomy and impact.
  • Competitive compensation, equity, and benefits in a rapidly scaling health tech company.

At Function, we celebrate diversity and are committed to building a diverse and inclusive workforce. As an equal opportunity employer, we do not discriminate on the basis of race, color, gender identity, ancestry, religion, age, sexual orientation, national origin, disability, marital status, Veteran status, or any other occupationally irrelevant criteria.
Join the Function Health team and become a part of our mission to build a healthier future for all. Discover more about us and how we're changing the face of healthcare atFunction Health


Apply now Apply later

* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰

Job stats:  5  0  0

Tags: APIs Architecture AWS CI/CD Computer Vision Data pipelines Distributed Systems Docker Engineering FastAPI Flask GCP Kubernetes LangChain LLMs Machine Learning Microservices ML infrastructure Model inference OpenAI Pipelines PostgreSQL Python RAG Reinforcement Learning Research Terraform

Perks/benefits: Career development Competitive pay Equity / stock options Health care Startup environment

Regions: Remote/Anywhere North America South America
Countries: Canada United States

More jobs like this