Machine Learning Engineer

Job Description

Join to apply for the Machine Learning Engineer role at Variational AI .

Small molecule drug discovery is one of the most exciting open problems in machine learning. Traditional approaches require over ten years and two billion dollars to develop a new pharmaceutical, and their reliance on trial-and-error calls out for better predictive and generative models. The existent datasets are large enough to benefit from sophisticated deep learning architectures, but small enough that ML models can be trained in a few days, facilitating rapid experimentation and innovation. Nevertheless, the current industry standard has progressed little beyond shallow ML techniques such as random forests and support vector machines, largely due to the difficulty of integrating world‑class machine learning research with chemistry and pharmacology expertise.

Variational AI is searching for a machine learning engineer to join us in our quest to radically accelerate the development of new drugs through machine learning excellence. For over six years, we have been advancing the state‑of‑the‑art, and delivering projects to customers including Merck, Rakovina Therapeutics, and ImmVue Therapeutics.

You will help run experiments to evaluate new architectures and tune hyperparameters; continually improve the robustness and efficiency of our existing code base; identify, clean, prepare, and test datasets; and apply our pipeline to new targets, to help develop novel drugs. In this process, you will have the opportunity to build your skills by collaborating with our team of accomplished ML scientists and chemists. Software engineering expertise is required, as well as experience with deep learning and generative models; knowledge of chemistry and pharmacology is preferred but not required.

Background We’re Looking For

• M.S. (Ph.D. preferred) in CS, applied mathematics, statistics, physics, or related discipline;
• Experience with machine learning techniques, including diffusion models, Transformers, and Bayesian optimization, preferably demonstrated through peer‑reviewed publications;
• Two or more years of experience developing robust code on larger projects, including code review, refactoring, unit testing, version control, etc.;
• Mastery of Python and PyTorch;
• Intellectual curiosity and drive to excel.

We are an equal opportunity employer and enthusiastically welcome applications from women, BIPOC, and members of underrepresented communities and groups.

Compensation is a competitive mix of cash and options. We prioritize expertise and passion over where you decide to live and work; however, for collaboration across our team, applicants must be based in North American time zones.

To learn more about us, you can find some of our recent work at variationalai.substack.com.

Seniority Level

Mid‑Senior level

Employment Type

Full‑time

Job Function

Engineering and Information Technology

#J-18808-Ljbffr