Job Details

Machine Learning Engineer, AI/ML for Drug Discovery

  2025-07-01     Atomwise     San Francisco,CA  
Description:

Machine Learning Engineer, AI/ML for Drug Discovery

Join to apply for the Machine Learning Engineer, AI/ML for Drug Discovery role at Atomwise

Machine Learning Engineer, AI/ML for Drug Discovery

Join to apply for the Machine Learning Engineer, AI/ML for Drug Discovery role at Atomwise

About Atomwise:

Atomwise is a TechBio company leveraging AI/ML to revolutionize small molecule drug discovery. The Atomwise team invented the use of deep learning for structure-based drug design; a core technology of Atomwise's best-in-class AI discovery and optimization engine, which is differentiated by its ability to find and optimize novel chemical matter. The company's belief is that structurally novel chemical matter increases the likelihood of developing first-in-class and best-in-class medicines that have the potential to transform patient care.

About Atomwise:

Atomwise is a TechBio company leveraging AI/ML to revolutionize small molecule drug discovery. The Atomwise team invented the use of deep learning for structure-based drug design; a core technology of Atomwise's best-in-class AI discovery and optimization engine, which is differentiated by its ability to find and optimize novel chemical matter. The company's belief is that structurally novel chemical matter increases the likelihood of developing first-in-class and best-in-class medicines that have the potential to transform patient care.

Atomwise has extensively tested its discovery and optimization engine, delivering hit ID success in over 230 academic and collaboration projects-to-date that cover a wide breadth of protein classes and numerous "hard-to-drug" targets. Atomwise is building a pipeline of small-molecule drug candidates with first-in-class and best-in-class potential in immunology.

The Role:

We seek a highly motivated and creative Machine Learning Engineer to join our growing team. You will be at the forefront of developing and applying advanced machine-learning models to accelerate the discovery of novel therapeutics. Your work will directly impact the development of innovative AI-driven drug discovery platforms. This role offers the opportunity to work on challenging problems at the intersection of machine learning, chemistry, and biology, with access to extensive datasets and state-of-the-art computational resources.

Responsibilities:

  • Productionize novel deep learning tools for molecular property prediction, design and optimization, and virtual screening.
  • Design and execute experiments to evaluate the performance of experimental models and improve training and inference time/cost.
  • Conduct in-depth analyses of model outputs and performance metrics to identify areas for improvement.
  • Collaborate with cross-functional teams of software engineers, chemists and biologists to integrate ML models into drug discovery pipelines.
  • Stay up-to-date with the latest advancements in machine learning and computational chemistry, and integrate them into Atomwise ML models.
  • Maintain and extend our scalable and efficient machine learning frameworks for at-scale use by drug discovery teams.

Qualifications:

  • Masters or higher in Computer Science, Machine Learning, Computational Chemistry, or a related field.
  • Two or more years of industry experience in productionizing machine learning models for distributed training and inference.
  • Proficiency in designing and implementing novel machine learning frameworks in PyTorch.
  • Strong problem-solving and analytical skills, with a demonstrated ability to conduct independent research.
  • Experience with collaborative software engineering and developing high quality, maintainable, well-documented code in a large codebase.
  • Experience with machine learning models of chemical systems is preferred.
  • Familiarity with cheminformatics tools and libraries (e.g., RDKit) is a plus.

Compensation & Benefits

  • Competitive salary, commensurate with experience
  • Stock compensation plan – you'll be an Atomwise co-owner
  • Platinum health, dental, and vision benefits for you and your dependents
  • 401(k) retirement plan with generous company match (up to 4%)
  • Flexible paid time off (PTO), 13 paid holidays, and wellness breaks for employees to spend time with their loved ones and recharge
  • Health Savings and Flexible Spending Account options to help save money on healthcare, daycare, and commuting
  • Employee Assistance Program (EAP) and Pet Insurance
  • Funding for professional development and conference attendance
  • Flexible work schedule
  • Generous paid parental leave

Atomwise is an equal opportunity employer and strives to foster an inclusive workplace. We are a TechBio company leveraging AI/ML to revolutionize small-molecule drug discovery, and we know that we need a diverse team to develop medicines that serve diverse populations. Accordingly, Atomwise does not make any employment decisions (including but not limited to, hiring, compensation, and promotions) on the basis of race, religion, color, national origin, gender, gender identity, sexual orientation, age, veteran status, disability status, or any other characteristics protected by applicable federal, state, and local law.

We strongly encourage people of diverse backgrounds and perspectives to apply.

The base salary for this position is $175,000- $260,000. Pay is determined by multiple factors including, but not limited to, a candidate's geographical location, experience and skills. We're considering all levels of experience for this position.

Seniority level

  • Seniority level

    Entry level

Employment type

  • Employment type

    Full-time

Job function

  • Job function

    Engineering and Information Technology

Referrals increase your chances of interviewing at Atomwise by 2x

Get notified about new Machine Learning Engineer jobs in San Francisco, CA

San Francisco, CA $135,000.00-$200,000.00 2 months ago

San Francisco, CA $170,000.00-$250,000.00 2 weeks ago

San Francisco, CA $130,000.00-$230,000.00 3 months ago

San Francisco, CA $140,000.00-$215,000.00 2 months ago

Research Engineer - Machine Learning (ML)

San Francisco, CA $100,000.00-$200,000.00 1 year ago

Redwood City, CA $123,000.00-$185,000.00 5 months ago

San Francisco, CA $140,000.00-$160,000.00 3 months ago

Machine Learning Researchers (Open-Endedness) - Open Level

Machine Learning Engineers (Open-Endedness) - Open Level

San Francisco, CA $130,000.00-$238,000.00 1 day ago

Machine Learning, Optimization. Remote or Hybrid

San Francisco, CA $150,000.00-$225,000.00 4 months ago

Machine Learning Engineer, Stripe Capital

San Francisco, CA $144,000.00-$247,500.00 5 hours ago

Sr. Software Engineer, ML Platform - Slack

San Francisco, CA $140,000.00-$180,000.00 3 months ago

San Francisco, CA $140,000.00-$160,000.00 2 months ago

San Francisco, CA $56.25-$173,000.00 2 weeks ago

Machine Learning Engineer, AI (FULLY REMOTE, USA)

San Francisco, CA $150,000.00-$200,000.00 4 months ago

San Francisco, CA $190,000.00-$320,000.00 2 months ago

San Francisco, CA $100,000.00-$180,000.00 1 year ago

San Francisco, CA $172,000.00-$384,100.00 2 days ago

San Francisco, CA $180,000.00-$270,000.00 4 weeks ago

Machine Learning Scientist, NLP (All Levels)

San Francisco, CA $200,000.00-$300,000.00 1 month ago

We're unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

J-18808-Ljbffr


Apply for this Job

Please use the APPLY HERE link below to view additional details and application instructions.

Apply Here

Back to Search