Role Description

Variational AI is searching for a machine-learning software engineer to join us in our quest to radically accelerate the development of new drugs through machine learning excellence. You will help improve our existing code base by:

• Improving memory and compute efficiency
• Developing new automation pipelines for preparing datasets and running experiments
• Moving research into production

In this process, you will have the opportunity to build your skills by collaborating with our team of accomplished ML scientists and chemists.

Qualifications

• Degree in CS, applied mathematics, statistics, physics, or related discipline
• Mastery of Python and PyTorch, including a deep understanding of tensor object structure and its performance implications, and the ability to write new, efficient tensor operations
• Experience in one or more of the following:
• Python extension modules
• Shell scripting
• Code profiling
• Large datasets, including storage formats and efficient loading
• Dataframe libraries, SQL
• Slurm or other HPC environments
• CUDA programming
• Cloud platforms
• Containers
• Two or more years of experience developing robust code on larger projects, including code review, refactoring, unit testing, version control, etc.
• Intellectual curiosity and drive to excel

Requirements

• Software engineering expertise is the primary requirement
• Knowledge of deep learning fundamentals
• Knowledge of chemistry and pharmacology is preferred but not required

Benefits

• Competitive mix of cash and options
• Equal opportunity employer, welcoming applications from women, BIPOC, and members of underrepresented communities and groups
• Prioritization of expertise and passion over location

Company Description

For over six years, Variational AI has been advancing the state-of-the-art and delivering projects to customers including Merck, Rakovina, and ImmVue Therapeutics. To learn more about us, you can find some of our recent work at variationalai.substack.com .