Software Engineer - AI Workflows

Posted 2 Days Ago
Hiring Remotely in New York, NY
Remote
Mid level
Healthtech • Insurance • Software
Your plan, powered by Flume
The Role
Develop AI workflows to interpret specifications, generate data mappings, and improve mapping accuracy using LLM APIs and orchestration tools.
Summary Generated by Built In
About Flume Health

Flume Health is a software company that connects the fragmented healthcare data ecosystem for more efficient health plan administration. As a single, cloud-native integration platform, Flume’s Relay platform allows companies to easily connect various systems and vendors for efficient data exchange that’s increasingly demanded of the modern health plan. Payers, third-party administrators, prescription benefits managers, and health solutions are provided the simplicity, speed, and security they need to automate data integration and movement between relevant stakeholders. Relay supports multiple data transmission protocols, data types, and file types. By streamlining data flows between payers and solutions, a world of opportunity exists to improve access to healthcare.


The Role:

At Flume Health, we’re leveraging AI to reimagine how healthcare data is mapped, validated, and integrated faster, more reliably, and at scale. We’re looking for a Software Engineer focused on AI workflows to help build and orchestrate pipelines that use foundation models like OpenAI’s or Google’s large language models (LLMs) to generate and iteratively refine data mappings.

You won’t be training models from scratch but you will be at the forefront of building robust workflows that interface with LLM APIs, run validation checks on outputs, and implement automated loops for refinement. If you enjoy piecing together modular AI processes, designing data-driven feedback loops, and working in a fast-paced, compliance-aware environment, this role is for you.

What You’ll Do:

  • Architect AI workflows that interpret user-provided specifications and generate data mappings using LLM APIs (e.g., OpenAI, Vertex AI), refining outputs based on validation feedback.
  • Implement intelligent retry and refinement loops to improve mapping accuracy using techniques like prompt chaining, structured validation, and contextual injection.
  • Develop prompt engineering strategies with support for domain-specific augmentation (e.g., via vector search or RAG).
  • Orchestrate pipelines with tools like Python and Airflow, integrating structured QA steps to monitor output quality and flag failures.
  • Collaborate across Data, Platform, and Product teams to embed these workflows into Flume’s core systems, ensuring performance, traceability, and compliance.
  • Stay up to date on the evolving LLM ecosystem understanding tradeoffs in cost, latency, and capability across vendors.

What You’ll Need:

  • 3+ years of software engineering experience, ideally in data, ML, or AI-driven product development.
  • Demonstrated experience building or orchestrating workflows that integrate with LLM APIs (e.g., OpenAI, Anthropic, Google).
  • Proficiency with Python and orchestration tools (e.g., Airflow, Dagster, Prefect).
  • Familiarity with prompt engineering concepts and iterative refinement techniques (e.g., looped retries, RAG, validation-based reruns).
  • Comfortable working in cloud-native environments (GCP or AWS) using Docker and Kubernetes.
  • Strong communication skills able to translate business needs into repeatable, explainable, and traceable AI workflows.

Technologies We Use:

  • LLM Tools & APIs: OpenAI, Vertex AI, DSPy, LangChain (where appropriate)
  • Workflow & Orchestration: Python, Airflow, custom feedback/retry frameworks
  • Cloud & Infra: GCP (BigQuery, Pub/Sub), Kubernetes, Docker, Terraform
  • Healthcare Standards: EDI, FHIR, HIPAA-compliant pipelines
  • Monitoring & Versioning: Logging/metrics pipelines, prompt version control

Nice to Haves:

  • Experience with vector databases or semantic search tools (e.g., Pinecone, Weaviate, Qdrant).
  • Exposure to regulated data environments, such as healthcare or finance.
  • Understanding of LLM orchestration frameworks (LangChain, DSPy, HuggingFace Accelerate).
  • Interest in evaluating the effectiveness of LLM outputs using custom metrics or domain-specific rules.
  • Contributions to open-source AI tooling or AI workflow libraries.

Special Note:

  • Candidates will be required to travel for in person team meetings once a quarter.

Salary Range:

  • $150K to $180K
Flume’s Perks & Benefits
  • Flexible PTO - you’re going to be working hard so enjoy time off
  • A robust stock option plan to give our employees a direct stake in Flume’s success
  • WFH stipend - we’ve always been remote first 
  • Competitive compensation and 401k with a 4% match
  • Comprehensive health coverage (medical, dental, vision) 

Top Skills

Airflow
AWS
Docker
Edi
Fhir
GCP
Hipaa
Kubernetes
Openai
Python
Vertex Ai

What the Team is Saying

Molly
Albana
Nathan
Cedric
Brian
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
22 Employees
Remote Workplace
Year Founded: 2018

What We Do

Flume Health is a software company that connects the fragmented healthcare data ecosystem for more efficient health plan administration. As a single, cloud-native integration platform, Flume’s Relay platform allows companies to easily connect various systems and vendors for efficient data exchange that’s increasingly demanded of the modern health plan. Payers, third-party administrators, prescription benefits managers, and health solutions are provided the simplicity, speed, and security they need to automate data integration and movement between relevant stakeholders. Relay supports multiple data transmission protocols, data types, and file types. By streamlining data flows between payers and solutions, a world of opportunity exists to improve access to healthcare.

Flume Health was founded in 2018, and closed a $30M series A in February 2022, led by Optum Ventures and CIGNA ventures—it was the fastest portfolio company that's ever gone through Optum's due diligence process. In total, Flume has raised over $40M from investors including Crosslink Capital, Accomplice VC, Primary VC, Founder Collective, Route 66, and Entrepreneurs Roundtable Accelerator (ERA).

Why Work With Us

At Flume Health, you’ll work alongside a multipurpose team of healthcare insiders and outsiders who are all committed to solving real-world problems. Everyone is an owner. We approach every problem assuming different stakeholders have different parts of the solution, and we need to collaborate to find the right answer.

Gallery

Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery

Flume Health Offices

Remote Workspace

Employees work remotely.

At our 100% remote company, employees have the flexibility to work from wherever they work best. For employees interested in meaningful in-person interaction, we host quarterly team gatherings and social events.

Typical time on-site: None
US
New York, NY
Learn more

Similar Jobs

Flume Health Logo Flume Health

Software Engineer

Healthtech • Insurance • Software
Remote
New York, NY, USA
22 Employees

Flume Health Logo Flume Health

Software Engineer - Data Platform

Healthtech • Insurance • Software
Remote
New York, NY, USA
22 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account