MLFF Distillation & GCMC Integration - Internship

CuspAI; Town of Cambridge; On-site; AUD 35,000 - 55,000; Full time

Benefits offered by this job

Competitive salary

Equity in CuspAI

Holiday leave

Parental leave

Professional development budget

Meaningful impact on materials science

Job summary

CuspAI is seeking an engineering intern to develop fast, accurate machine learning force fields (MLFFs) for MOF screening and integrate them into kUPS. You will work within the chemistry team and contribute to end-to-end ML-driven simulations.

Ideal candidates are enrolled in or recently completed a PhD or Master's in a quantitative field, with experience in adsorption modelling, molecular simulations (GCMC/MD), and Linux-based workflows.

Qualifications

• Currently enrolled in (or recently completed) a PhD or Master's programme in a relevant quantitative field (Physics, Chemistry, Chemical Engineering, Computational Science, Machine Learning, or similar).
• Experience in adsorption modelling at the atomic scale.
• Hands-on experience with molecular simulation methods (GCMC, MD, or both).
• Comfortable working on Linux environments and managing simulation campaigns at scale.
• A genuine interest in the application of ML to chemistry and materials science.

Responsibilities

• Distill MLFFs into fast student potentials optimised for Monte Carlo simulations.
• Curate, version, and document training and validation datasets, including the distillation protocol and any active-learning loops used to close coverage gaps.
• Run head-to-head validation campaigns comparing the distilled MLFF against classical force-field baselines across a curated set of guest molecules, characterising accuracy, throughput, and failure modes.
• Profile and optimise the pipeline for throughput, with particular focus on the MC inner loop where MLFF inference cost dominates.
• Collaborate with computational chemists on reference data generation, benchmark system selection, and validation strategy.

Skills

ML in chemistry

Molecular simulation

Python

Linux

Education

PhD or MSc in a quantitative field

Tools

GCMC

MD

kUPS

Active learning

Job description

About CuspAI

CuspAI is the frontier AI company on a mission to solve the breakthrough materials needed to power human progress. While nature took billions of years to perfect molecules, we are harnessing AI to unlock trillion-dollar materials breakthroughs in months, not millennia. Our founding team is the most cited in the world, comprised of world-class researchers in AI, chemistry and engineering. We are working on some of the hardest and most important challenges including energy, clean water, the future of compute, and carbon capture, and this is just the start of what our 'search engine' for next-generation materials will unlock. We invite you to be part of a diverse, innovative team at the intersection of AI and materials science, working to create impactful partnerships that drive innovation, scalability, and industry collaboration. This work matters. Your work matters. We're on the cusp of the on-demand materials era. Join us.

The Role

We are seeking an intern for a 3-month internship to develop fast, accurate machine learning force fields (MLFFs) tailored to high-throughput Monte Carlo simulation, and integrate them into our in-house simulation framework, kUPS. You will be embedded in our chemistry team and work closely with CuspAI colleagues.

Note: You would be joining as an engineering intern within the chemistry team at CuspAI.

Your Impact

You will deliver one of the foundational capabilities our simulation stack needs to evaluate the next generation of MOFs: an MLFF that is both accurate enough to replace classical force fields for guest-host interactions and fast enough to run inside the inner loop of GCMC. By distilling state-of-the-art equivariant models into a lightweight student potential and integrating the result directly into kUPS, you will expand what is computationally tractable for CuspAI and the wider gas adsorption community.

What You Will Do

Models

• Distill MLFFs into fast student potentials optimised for Monte Carlo simulations.




• Curate, version, and document training and validation datasets, including the distillation protocol and any active-learning loops used to close coverage gaps.

Integration & Validation

• Run head-to-head validation campaigns comparing the distilled MLFF against classical force-field baselines across a curated set of guest molecules, characterising accuracy, throughput, and failure modes.

Systems & Infrastructure

• Profile and optimise the pipeline for throughput, with particular focus on the MC inner loop where MLFF inference cost dominates.




• Benchmark accuracy/speed trade-offs systematically and document where the distilled model fails.

Science & Collaboration

• Collaborate with computational chemists on reference data generation, benchmark system selection, and validation strategy.




• Contribute to a publication establishing MLFF-driven GCMC for MOF screening.

Must Have Skills and Qualifications

• Currently enrolled in (or recently completed) a PhD or Master's programme in a relevant quantitative field (Physics, Chemistry, Chemical Engineering, Computational Science, Machine Learning, or similar).




• Experience in adsorption modelling at the atomic scale.




• Hands-on experience with molecular simulation methods (GCMC, MD, or both).




• Comfortable working on Linux environments and managing simulation campaigns at scale.




• A genuine interest in the application of ML to chemistry and materials science.

Bonus Points (But Not Critical)

• Familiarity with modern MLFFs.




• Experience with knowledge distillation or other model compression techniques for scientific ML.




• Experience with active learning workflows for atomistic data.




• Familiarity with DFT data generation and the practicalities of curating atomistic datasets.




• Direct experience with established simulation packages.




• Background in gas adsorption, MOFs, or porous materials.




• Familiarity with classic force fields used in MOF simulation.




• A track record of published research at top-tier ML or computational chemistry venues.

What We Offer

• A competitive salary: We value and reward impact and growth




• Equity in CuspAI: You have a stake in the success of the company




• Time off to stay fresh: 28 days holiday (DE, NL, UK) or 21 days holiday (JP, SG, US), in addition to local public holidays




• 'Gold Standard' parental leave: 26 weeks (primary caregiver) and 12 weeks (secondary caregiver) at full pay - we look after you and your family while we work on the most important materials discovery problems together




• Professional development budget: We invest in your career development so you can stay up to date with the latest industry knowledge or add to your skills to increase impact and growth




• Solve meaningful problems: See how your work has a direct impact on advancing materials science and solving sustainability and climate-related problems through the creation and application of bleeding-edge SOTA technology and revolutionary techniques




• True interdisciplinary teamwork: Be part of a deeply collaborative environment bridging AI research, computational chemistry, and experimental science - work with world-class researchers and engineers who enjoy sharing knowledge and supporting each other

Join us in shaping the future of materials with AI. Together, we can create groundbreaking solutions for a more sustainable world. CuspAI is an equal opportunities employer committed to building a diverse and inclusive workplace. We do not discriminate on the basis of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy or related condition (including breastfeeding), veteran status, or any other basis protected by applicable law. We actively encourage applications from all backgrounds and value the unique perspectives and contributions that diversity brings to our team. Please let us know if you require any specific adjustments during or after the interview process. We will do everything we can within reason to accommodate.