AI Engineer

Max AI, Inc.; California; On-site; USD 150,000 - 250,000; Full time

Benefits offered by this job

Equity

Platinum Healthcare

Gym stipend

Job summary

Max AI, Inc. is building the world’s first human-free, fully autonomous medical billing AI agent, starting with dermatology.

The company claims a $350B/yr market and highlights a track record of rapid expansion and notable leadership, including experts from MIT/Caltech and Figma. The AI Engineer role focuses on developing agents and ML models for accurate medical coding, collaborating closely with Operations and customers to ship practical AI capabilities with strong ownership and an agile,

Qualifications

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

Responsibilities

• Build, experiment, and evaluate AI agents and ML models in the NLP domain to improve medical coding/billing.
• Contribute to internal AI frameworks enabling scalable AI agents and ML models.
• Collaborate with a small team and engage with Operations and customers to drive improvements.

Skills

ML in production

NLP

Python ML libraries

LLMs

Transformer models

Model evaluation

Tools

TensorFlow

PyTorch

scikit-learn

Transformers

HuggingFace

AWS

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)