AmazonIN, TS, Hyderabad
Cusp AIPosted 2mo ago
MLFF Distillation & GCMC Integration - Internship
MLFF Distillation & GCMC Integration - Internship at Cusp AI scores 88 out of 100 on AI centrality, which makes it a Level 4 role on this board.
AI in this role
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 classical 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.
Prepare for this job
A free preview built only from this posting: what it asks for, what you could be asked in an interview, and how to adjust your resume.
Skills and AI tools this role asks for
Questions you could be asked
- Tell me about a research question you investigated. What did you find?
- How would you decide a model or AI system is ready to ship?
- Tell me about a time a model underperformed in production. How did you find out, and what did you change?
Adapt your resume
- List these exact terms on your resume: AI Research. An applicant tracking system matches the wording, not the idea.
- Attach one line of real, concrete experience to at least one of them — a tool named with nothing behind it rarely survives a human read.
- Lead with what you built, trained or shipped — this role is judged on the AI system itself, not the tools around it.
Want your resume actually rewritten for this job?
The free preview above is everything we have today. A full resume rewrite is not live yet and has no price set. Join the waitlist and we will email you if we open it.
Similar roles
Other roles rated Level 4 at other companies.



