Machine Learning Engineer

The company is a kidney care management company that partners with physicians on its mission to reimagine healthcare—with the expertise, scale, compassion, and vision to set the standard for the industry and help patients live their best lives. We are on a mission to help people and we know the work we do changes their lives. If there is a better way, we will create it. So, if our mission speaks to you, join us!

As a Machine Learning Engineer, you’re a highly motivated individual with strong fundamentals in computer science and hands‑on experience across the full model development lifecycle—including feature engineering, model development, calibration, deployment, and ongoing monitoring. In this role, you will be flexible, eager to learn new skills, and willing to contribute wherever the team needs support. This Machine Learning Engineer is comfortable working with both traditional tabular machine learning models and modern AI techniques, including prompt engineering and LLM‑based capabilities.

What You’ll Do

• Develop and deliver end‑to‑end machine learning solutions, including defining technical requirements, architecting scalable systems, and implementing monitoring, logging, and maintenance workflows.

• Collaborate closely with engineers, product managers, clinicians, and cross‑functional partners to build new ML products and enhance existing systems.

• Lead the design and implementation of MLOps frameworks, including pipeline development, CI/CD integration, drift detection, retraining workflows, and rollback strategies.

• Monitor model performance in production, identify issues, propose remediation steps, and ensure strong test coverage and system reliability.

• Utilize contemporary software engineering practices to implement scalable, secure, and maintainable AI/ML systems.

• Develop and customize API integrations to enable seamless connectivity between cloud‑based systems and ML services.

• Participate in architectural discussions to ensure ML platforms meet compliance, performance, and scalability standards.

What You’ll Need

• Bachelor’s degree in Computer Science, Data Analytics, Software/Computer Engineering, Computational Statistics, Mathematics, or a related discipline.

• 3+ years of end‑to‑end ML development in production (data prep, feature engineering, modeling, calibration, deployment, monitoring, maintenance).

• 3+ years of MLOps experience building production pipelines (CI/CD, model registry, feature store), implementing monitoring & drift detection, and automating retraining.

• 3+ years of Python for production ML (testing, packaging, type hints, linting) and SQL for analytical and production workloads; Scala a plus.

• 2+ years working with distributed compute and cloud ML environments (e.g., Spark/Databricks on Azure/AWS/GCP) and modern data ecosystems (data lakes, DBMS).

• Strong debugging and optimization skills across data and ML workflows.

• Track record of ownership and problem solving—driving measurable impact and quality under ambiguity and evolving requirements.

• Ability to communicate technical decisions clearly and contribute to documentation and design discussions.

• Demonstrated system design & architecture skills for scalable, high‑performance ML services and batch/streaming workflows; familiarity with API design and service integration patterns.

• Proven understanding of tradeoffs in latency, cost, performance, and compliance.

Preferred

• 1+ years of Databricks experience + some experience in infrastructure/networking

• 1+ years implementing LLM‑based solutions in production (prompt/response design, evaluation frameworks, guardrails/safety, latency/cost optimization).

• 1+ years designing compliant ML platforms (e.g., HIPAA, SOC 2) and working with PHI/PII governance, access controls, and auditability.

Our mission is to reinvent healthcare to help patients live their best lives, and we proudly live our mission-driven values:

• We care deeply about the people we serve.

• We are better when we work together.

• Humility is a source of our strength.

• We bring joy to our work.

• We deliver on our promises.

We are committed to diversity, equity, and inclusion throughout our recruiting practices. Everyone is welcome and included. We value our differences and learn from each other. Our team members come in all shapes, colors, and sizes. No matter how you identify your lifestyle, creed, or fandom, we value everyone's unique journey.

Oh, and one more thing … a recent study shows that men apply for a job or promotion when they meet only 60% of the qualifications, but women and other marginalized groups apply only if they meet 100% of them. So, if you think you’d be a great fit, but don’t necessarily meet every single requirement on one of our job openings, please still apply. We’d love to consider your application!

Come join us and help our patients live their best lives. Learn more at [link for members]

It has come to our attention that some individuals or organizations are reaching out to job seekers and posing as potential employers presenting enticing employment offers. We want to emphasize that these offers are not associated with our company and may be fraudulent in nature. Please note that our organization will not extend a job offer without prior communication with our recruiting team, hiring managers and a formal interview process.