Senior Machine Learning Engineer

At Unlearn, our purpose is to advance artificial intelligence (AI) to eliminate trial and error in medicine. We are innovating advanced machine learning methods to leverage generative AI in forecasting patient outcomes, starting with the domain of clinical trials. We produce AI-generated digital twins of individual trial participants, enabling smaller and more efficient clinical trials to bring effective medicines to patients sooner.

Our innovative work in AI today will reinvent how AI is applied in medicine tomorrow — and we have a top secret plan for how to get there. We won’t be able to achieve this mission just by applying technologies created by others; the future must be invented.

Unlearn is a technology company, not a biotech company. We use computers, not pipettes. We make and use software, we don’t discover or make drugs. We believe that AI will define the future of medicine, and we aren’t deterred by naysayers or skeptics. 

We come from a variety of backgrounds ranging from machine learning to marketing—but regardless of where we come from, Unlearners share some common traits:

• Unlearners are ambitious; we aren’t intimidated by big, challenging goals. 

• Unlearners are disciplined experimenters; we break down our big goals into smaller chunks and meet as often as necessary to track our velocity and iterate quickly. 

• Unlearners are gritty; we never give up, setbacks just make us try harder.

• Unlearners are receptive to new ideas; in fact, we hate being stuck with the status quo

• Unlearners are storytellers; sharing information with each other and with the world is super important, too important to be boring.  And, last but not least,

• Unlearners are team-oriented; we put the mission first, the company second, the team third, and individuals last.

Headquartered in San Francisco, Unlearn was founded in 2017 by a team of world-class machine learning scientists.  We have raised venture capital from top tier investors such as Altimeter, Insight Partners, Radical Ventures, 8VC, DCVC, and DCVC Bio, and recently completed our $50 million Series C in January 2024.

If our purpose and culture resonate with you, we invite you to apply.

Machine Learning Engineers innovate on Unlearn’s approach to build state-of-the-art ML systems for generating Digital Twins – probabilistic models of a patient’s future health outcomes given knowledge of their current and past medical history. ML Engineers at Unlearn come from a wide range of disciplines, and have honed their craft through their experience delivering robust and reliable ML products and systems. Successful ML Engineers at Unlearn are entrepreneurial in their approach; feeling a strong sense of end-to-end ownership of their mission, they investigate broadly to find the right tools and techniques to help their teams succeed. They are also highly determined individuals, powering through problems with cleverness and resolve.

Responsibilities include:

• Design and implement machine learning models to characterize and predict disease progression.

• Implement, evaluate, and fine-tune novel model architectures as part of ongoing innovation.

• Translate conceptual ML approaches into reliable and performant codebases.

• Improve and optimize ML infrastructure, focusing on training and inference efficiency.

• Develop and maintain software libraries to support scalable ML workflows.

• Measure, analyze, and report on the health of deployed ML systems and products.

• Track state-of-the-art with respect to ML engineering tools and approaches.

Minimum requirements:

• M.S. in computer science or engineering, physics, mathematics, or a related field.

• Strong software engineering skills and collaborative software development.

• Fluency with Python and its software ecosystem.

• Solid fundamentals in conceptual basics of ML architecture (linear algebra, statistics, optimization).

• Deep experience with ML frameworks such as PyTorch and/or TensorFlow.

• Experience with high-performance and/or distributed computing environments.

• Demonstrated history of delivering production ML systems or impactful research tools.

• Hands-on experience with influential ML architectures of the last 5 years (e.g. transformers, diffusion, etc.)

Bonus points for:

• Ph.D. or equivalent industry research experience.

• Experience directly bearing on Unlearn’s core domain: generative modeling of tabular time-series data.

• A record of publication contributions which used systems you developed to deliver impactful research results.

• Contributions to well-known open-source ML tools or frameworks.

• Proficiency with AWS cloud services.

• Prior experience working with healthcare or clinical machine learning applications.

• Full-ML-stack experience: delivering ML services from conception to production system to maintenance and iterative improvement.

Benefits & Perks

The following benefits and perks are for full time roles only.

• Generous equity participation

• 100% company-covered medical, dental, & vision insurance plans

• 401k plan with matching

• Flexible PTO plus company holidays

• Annual company-wide break December 24 through January 1

• Commuter benefits

• Paid Parental Leave

• Support for H1B, TN, and E-3 Visa change of employer transfers

Unlearn is an equal opportunity employer. 

At Unlearn, we are committed to building a diverse and inclusive workplace, because inclusion and diversity are essential to achieving our mission. If you’re excited about this role, and your past experience doesn’t align perfectly with every qualification in the job description, we encourage you to apply nevertheless.