# Staff Software Engineer (AI)

**Company:** [Onos Health](https://hotfix.jobs/companies/onos-health)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Salary:** $180k – $250k/yr
**Experience:** 5+ years
**Skills:** Python, Django, LLMs, AWS, Postgres, Docker, Celery, S3, MLOps, HIPAA
**Posted:** 2025-12-17

> Develops LLM/NLU systems and data pipelines to process medical records, classify patients, and recommend care levels for a healthcare AI platform. Requires 5+ years backend/data engineering experience with production LLMs and integration into user-facing features.

## Job Description

## Responsibilities
- Develop LLM/NLU systems to process and extract meaningful information from clinical notes and medical documents, classify patients according to level-of-care guidelines, and make accurate recommendations.
- Establish best practices for LLM/AI systems, benchmarking/evaluation frameworks, and model governance to improve reliability, maintainability, and scalability.
- Build data pipelines that scale efficiently while maintaining strict data privacy and security standards.
- Collaborate with backend engineers to integrate AI/ML capabilities seamlessly into the Onos platform.

## Requirements
- 5+ years experience building and deploying applications in production in a backend engineering / data engineering capacity.
- Relevant experience with developing LLM-based systems for ingesting and evaluating unstructured records for industry-specific use cases and integrating them with user-facing features.
- Deep understanding of the limitations of using LLMs and the best practices for using them for reliable, consistent, and accurate outputs.
- Customer obsessed and motivated to build a best-in-class model for behavioral health clinical assessments in the healthcare space.
- A collaborative team player with a focus on delivering measurable results.

## Nice-to-Haves
- Specifically worked with medical records to evaluate whether a patient’s history meets criteria for evaluations or assessments (e.g., claims authorization or other types of evaluations).
- Experience wearing multiple hats as a generalist backend engineer.
- Experience working with data pipelines and Python and related data science/ML libraries.
- Significant experience working with healthcare data and with HIPAA best practices.
- Knowledge of modern LLM and ML infrastructure and MLOps best practices.

## Tech Stack
**Infrastructure/Systems:** AWS (ECS, Bedrock, Cognito, etc.), Docker, Github Actions
**Languages/Frameworks:** Python, Django, Celery, django-ninja, django-tenants
**Database/Storage:** PostgreSQL (AWS RDS), S3
**Development Tools:** Github, Jira, Coderabbit, Tusk, Claude

## Benefits
- Flexible hybrid arrangement: 2-3 days/week at San Francisco office (Financial District), remote-first culture.
- Unlimited vacation policy.
- Paid parental leave.
- Medical, dental, and vision insurance.
- Pre-tax commuter benefits.
- 401(k).
- Significant equity as an early employee.
- Direct mentorship from experienced founders.
- Ground-floor opportunity to help build a team and culture.
- Regular team events and offsites.
- Company-provided equipment and home office setup.

## Similar roles

- [Member of Technical Staff](https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff, Applied Research](https://hotfix.jobs/jobs/d19460df-bf56-4f87-aa95-5e7bdf40798f) - LlamaIndex - San Francisco, CA - $180k – $250k/yr
- [Member of Technical Staff, Inference](https://hotfix.jobs/jobs/70280e8d-9e99-4be1-a488-b13edc2d4080) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff](https://hotfix.jobs/jobs/544a5b36-3a56-45fa-89fe-0873120ece22) - Black Forest Labs - San Francisco, CA - $180k – $290k/yr
- [Member of Technical Staff - X Search](https://hotfix.jobs/jobs/2480c779-d772-4053-b432-64b3e48d4cd3) - xAI - Palo Alto, CA - $180k – $440k/yr

**Apply:** https://hotfix.jobs/jobs/f4803284-f70b-45d6-9077-15ccbe640291
**Canonical:** https://hotfix.jobs/jobs/f4803284-f70b-45d6-9077-15ccbe640291