Sr. Machine Learning Engineer

AKASA; South San Francisco (CA); Remote; USD 175,000 - 230,000; Full time

Job summary

AKASA, a leading provider of generative AI solutions in healthcare, seeks a Sr. Machine Learning Engineer. Work remotely with a talented team, develop ML solutions for healthcare challenges, and leverage your expertise to empower healthcare professionals. Enjoy competitive compensation, unlimited PTO, and a commitment to diversity and inclusion.

Benefits

Unlimited paid time off (PTO)

Expansive coverage for health, dental, and vision

Employer contribution to Health Savings Accounts (HSA)

Generous parental leave policy

Full employee coverage for life insurance

Company-paid holidays

401(K) plan

Qualifications

• 5+ years of work experience in machine learning and data engineering.
• Experience launching production systems from the ground up.
• Development experience with cloud platforms such as Spark, AWS & k8s.

Responsibilities

• Develop state-of-the-art ML solutions for healthcare problems.
• Write production-ready software with well-tested algorithms.
• Own ML services end-to-end, from problem discovery to deployment.

Skills

Machine Learning

Data Engineering

Python

C++

Cloud Platforms

ML Frameworks

Full-stack Development

Large Language Models

Education

Master's degree in Computer Science or similar

Tools

AWS

Spark

k8s

Scikit-Learn

Pytorch

TensorFlow

Job description

About AKASA

At AKASA, our mission is to build the future of healthcare with AI. As the preeminent provider of generative AI solutions for the healthcare revenue cycle, we are solving the biggest challenges facing our customers’ financial infrastructure. We have raised more than $205M in funding from investors such as Andreessen Horowitz, BOND, and Costanoa Ventures.

This is one of the most exciting times to join AKASA. Bookings for our new product suite have grown 10x since launching in Q3 of 2024, and we are just getting started. Our customer base represents more than $120B+ in net patient revenue and includes the most innovative health systems in the country, like Stanford, Johns Hopkins, and Cleveland Clinic.

Some of our most recent recognitions include being named one of the fastest-growing GenAI startups to watch by AIM Research and an “AI Revenue Cycle Leader” by Black Book in 17 out of 18 categories. Our CEO was ranked among the “Top 50 Healthcare Technology CEOs” by the Healthcare Technology Report, and we have been certified as a “Great Place to Work” for the past five years in a row, just to name a few.

Come join us - everyone is welcome. As an inclusive workplace, we are committed to building an environment where AKASAns are comfortable bringing their authentic selves to work.

About the Role

As a remote Sr. Machine Learning Engineer, you'll report to our Engineering Manager and work with a talented team of PhD Researchers, ML Engineers, and healthcare experts. You will put machine learning into practice, so your code directly affects our customers immediately. You’ll work with large proprietary medical and clinical datasets containing both structured documents, natural language and images. Our mission is to empower healthcare professionals with tools that amplify their capabilities, making them faster, more comprehensive, and more effective in their roles.

The AKASA office is located in South San Francisco. While we support remote work on a variety of teams, we have a strong Bay Area presence across the company. The local R&D teams come into the office every Wednesday for co-working days, which this role will be expected to attend.

What You'll Do

• Participate in developing state-of-the-art ML solutions to address large-scale healthcare problems

• Help develop pipelines that collect, preprocess, and deliver data with a measurable quality

• Write production-ready software with well-tested and efficient algorithms

• Develop state-of-the-art ML algorithms across computer vision, large-language models, and probabilistic inference to solve problems like medical document entity extraction, medical coding and claim outcome prediction

• Own ML services end-to-end, including problem discovery, data pipeline development, inference optimizations, model experimentation, and service deployment

• Help build novel, application-specific ML models - all of our products are built from the ground up with ML at their core, enabling us to deploy our predictions in new and interesting ways

Skills & Qualifications

• Master's degree in Computer Science or similar

• 5+ years of work experience in machine learning and data engineering

• Have experience launching production systems from the ground up

• Proficiency in one or more programming languages such as Python and C++

• Development experience with cloud platforms such as Spark, AWS & k8s

• Knowledge of ML frameworks such as Scikit-Learn, Pytorch & TensorFlow

• Full-stack development experience for an end-to-end ML solution

• Ideal experience with Large Language Models, including both open source model deployment and agentic workflows.

What We Offer

• Unlimited paid time off (PTO)

• Expansive coverage for health, dental, and vision

• Employer contribution to Health Savings Accounts (HSA)

• Generous parental leave policy

• Full employee coverage for life insurance

• Company-paid holidays401(K) plan

Compensation

• Based on market data and other factors, the salary range for this position is $175,000-$230,000 + Equity. However, a salary higher or lower than this range may be appropriate for a candidate whose qualifications differ meaningfully from those listed in the job description.

The above represents the expected salary range for this job requisition. Ultimately, in determining your pay, we'll consider your location, experience, and other job-related factors.

We’re committed to doing the best work of our lives, together. Come see if we're the right team for you.

AKASA is a proud equal opportunity employer and we believe that a diverse and inclusive workforce is an imperative. We welcome people of different backgrounds, genders, races, ethnicities, abilities, sexual orientations, and perspectives, just to name a few. We do not discriminate based upon any protected class and we encourage candidates of all identities and backgrounds to apply. AKASA considers qualified applicants regardless of criminal histories in accordance with the San Francisco Fair Chance Ordinance.

AKASA is committed to providing reasonable accommodations for candidates with disabilities in our recruiting process. If you need any assistance or accommodations due to a disability, please let us know at [e-mail].