AI Engineer

Max AI; San Francisco (CA); On-site; USD 150,000 - 250,000; Full time

Benefits offered by this job

$250 gym membership

$2,000 annual courses/mentorship

Job summary

Max AI – Stripe for Healthcare is seeking an AI Engineer to advance autonomous medical billing through NLP and ML. The role will build, evaluate, and deploy AI agents and ML models, and collaborate with Operations and customers to drive meaningful improvements in revenue-cycle automation.

The position requires 6+ years in production ML/AI, deep NLP expertise, and hands-on experience with LLM fine-tuning, TensorFlow, PyTorch, and transformer-based tools.

Qualifications

• 6+ years of ML/AI in production.
• Proficient with ML/NLP Python packages (TensorFlow, PyTorch, scikit-learn, transformers).
• Expertise in AI/ML, NLP, embeddings and transformer models.
• Experience with LLMs, supervised fine-tuning, AI frameworks and evaluating LLMs.

Responsibilities

• Build, experiment, and evaluate AI agents and ML models in NLP for medical coding/billing.
• Contribute to internal AI frameworks to scale AI agents and models.
• Collaborate with Operations and customers to drive product improvements.

Skills

ML in production

NLP expertise

Python ML tools

LLM fine-tuning

Tools

TensorFlow

PyTorch

scikit-learn

Transformers

Job description

Max AI – Stripe for Healthcare

Max AI is the World's first human-free, fully-autonomous medical billing AI agent. Many startups are attempting to attack this problem because the market is so big — $350B/yr. We have the best team going after it. While everyone else still needs humans in the loop, we've cracked the code for medical billing starting with dermatology, and are rapidly adding support for every other specialty. Medical doctors who code are rare. We have the best one in the world. His previous apps have driven 50M+ downloads, all while running an incredibly successful practice of his own where he literally did the billing by hand himself. Our Head of AI has been doing AI research at MIT and Caltech for over 10 years. And our Head of Engineering was one of the earliest engineers at Figma.

AI Engineer

Responsibilities

• Build, experiment, and evaluate AI agents and ML models in the NLP domain to improve the accuracy of medical coding/billing and other aspects of revenue cycle management
• Contribute to the building of internal AI frameworks to enable the building of our AI agents and ML models
• Collaborate closely with a small, high-impact team and engage directly with our Operations team and our customers to drive meaningful improvements

Work Approach

• Complete product ownership: tackle challenges across the board rather than working in silos
• An agile, iterative process with clear accountability
• Regular interactions with customers to ensure our product continuously evolves to meet their needs

Core Values

• A proactive mindset: Spot and resolve issues, continuously driving improvements
• Commitment to lifelong learning and personal growth
• Positivity, collaboration, and a team-first attitude
• Clear communication skills

Requirements

Required

• 6+ years of ML/AI/data science in a production setting
• Well-versed in using ML/NLP python packages such as tensorflow, pytorch, scikit-learn, transformers (including others for working with huggingface models)
• Expertise in AI/ML and specifically in NLP (pre-processing, embeddings, transformer-based models, etc)
• Experience with LLM, including supervised fine-tuning, usage of AI frameworks, evaluations of LLMs

Nice to Have

• Experience working at high-growth startups
• Exposure in highly regulated industries and/or healthcare
• Experience with AWS

Benefits

Compensation

• $150k - $250k USD
• Equity

Platinum Healthcare

• Medical (Platinum PPO) – Company covers 100% for employee, 80% for dependents
• Dental (Platinum PPO) – Company covers 90% for employee, 75% for dependents
• Vision (Platinum PPO) – Company covers 90% for employee, 75% for dependents

Perks

• $250 per month (gym membership, classes, etc)
• $2,000 annually (courses, mentorship, etc)