Staff Software Engineer (AI/ML) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Onos Health · 3 weeks ago

Staff Software Engineer (AI/ML)

Onos Health is a company dedicated to improving healthcare efficiency through AI-driven solutions. The Staff Software Engineer (AI/ML) will be responsible for developing machine learning models and data pipelines to enhance healthcare data analysis and ensure high-quality care delivery.

Health CareSaaSSoftware
check
H1B Sponsor Likelynote

Responsibilities

Design, develop and deploy sophisticated ML models to analyze healthcare data and detect anomalies, classify patients according to level-of-care guidelines, and make accurate recommendations
Develop LLM/NLU systems to process and extract meaningful information from clinical notes and medical documents
Build data pipelines that scale efficiently while maintaining strict data privacy and security standards
Establish machine learning & AI best practices, evaluation frameworks, and model governance for future team growth
Collaborate with backend engineers to integrate AI/ML capabilities seamlessly into the Onos platform
Build AI/ML pipelines to analyze medical records to streamline clinical assessments and healthcare quality reviews
Design explainable AI solutions that provide transparency into model decisions for healthcare professionals
Develop an intelligent engine that ingests complex medical standards of care documents and evaluates provider adherence to guidelines
Create robust anomaly detection algorithms to identify patterns of fraud, waste, and abuse in healthcare claims

Qualification

Machine LearningPythonData PipelinesLLM/NLU SystemsExplainable AIHealthcare DataModel GovernanceMLOpsCollaborationCustomer Focus

Required

5+ years experience building and deploying machine learning systems in production
Significant experience working with data pipelines and Python and related data science/ML libraries
Experience with developing LLM-based systems 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 make an impact in the healthcare space
A collaborative team player with a focus on delivering measurable results

Preferred

Experience wearing multiple hats as a generalist backend engineer
Significant experience working with healthcare data and with HIPAA best practices
Experience with explainable AI and model governance in regulated industries
Led teams of ML engineers or data scientists on major projects
Knowledge of modern ML infrastructure and MLOps best practices
Built and deployed large language models for production applications

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

Company

Onos Health

twittertwitter
company-logo
Onos Health is a health tech company that offers clinical teams a behavioral health platform to streamline workflows and cut admin tasks.

H1B Sponsorship

Onos Health has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)

Funding

Current Stage
Early Stage
Total Funding
$6.32M
2025-05-06Seed· $6.32M
Company data provided by crunchbase