StrainRelief calculates the ligand strain of uncharged docked poses and has a suite of different force fields with which to do this. This includes a MACE neural network potential trained on SPICE2.
- 📄 The publication can be found here.
- 📊 All relevant datasets here.
- 💬 RAG chatbot for questions about the paper and references.
- 💻 Chatbot source code.
- Inclusion of NNP ASE calcualtors is more modular, making it easier to add your own.
- Meta's FairChem e-SEN NNP from OMol25 has been added giving a significant performance boost! Requested access here.
- Improved hydra configurations mean
model.model_pathsnow only has to be specified once. - Our paper has been published in the Journal of Chemical Information and Modelling!
- We have written a RAG chatbot to answer questions about the code, paper and any of its references.
From the root directory, run the following commands to install the package and its dependencies in editable mode:
(mace-torch==0.3.x requires e3nn==0.4.4 (only for training, not inference). fairchem-core requires e3nn>=0.5. So until mace-torch==0.4 is released we will have to do this finicky way of installing [GitHub issue])
mamba env create -f env.yml
mamba activate strain
pip install -e .
pip install --force-reinstall e3nn==0.5 fairchem-core
pre-commit installInstall uv and from the root directory run:
curl -LsSf https://astral.sh/uv/install.sh | sh
uv syncThe protocol used in StrainRelief is designed to be simple, fast and model agnostic - all that is needed to apply a new force field is to write an ASE calculator wrapper. Additionally you can use any MACE model, such as these from the MACE-OFF23 repository.
The protocol consists of 5 steps:
- Minimise the docked pose with a loose convergence criteria to give a local minimum.
- Generate 20 conformers from the docked ligand pose.
- Minimise the generated conformers (and the original docked pose) with a stricter convergence criteria.
- Evaluate the energy of all conformers and choose the lowest energy as an approximation of the global minimum.
- Calculate
E(ligand strain) = E(local minimum) - E(global minimum)and apply threshold.
N.B. energies returned are in kcal/mol.
Choose a minimisation and energy evalation force field from mmff94, mmff94s, mace, esen.
The calculator works best when the same force field is used for both methods. If this is the case, energy_eval does not need to be specified.
See the example scripts in examples along with a tutorial to explain StrainRelief's output and some handy helper functions.
This is the simplest and fastest implementation of StrainRelief using MMFF94s and a minimial example dataset.
strain-relief \
io.input.parquet_path=data/example_ligboundconf_input.parquet \
io.output.parquet_path=data/example_ligboundconf_output.parquet \
minimisation@local_min=mmff94s \
minimisation@global_min=mmff94s \
local_min.fmax=0.50
This script demonstrates using different force fields for minimisation (MMFF94s) and energy evaluations (MACE).
strain-relief \
io.input.parquet_path=data/example_ligboundconf_input.parquet \
io.output.parquet_path=data/example_ligboundconf_output.parquet \
minimisation@local_min=mmff94s \
minimisation@global_min=mmff94s \
energy_eval=mace \
model=mace \
model.model_paths=models/MACE_SPICE2_NEUTRAL.model
This is the script as used for most calculations in the StrainRelief paper. MACE is used for minimisation (and energy evalutions implicitly). A looser convergence criteria is used for local minimisation. Note: a gpu is required by default to run calculations with MACE.
strain-relief \
io.input.parquet_path=data/example_ligboundconf_input.parquet \
io.output.parquet_path=data/example_ligboundconf_output.parquet \
minimisation@global_min=mace \
minimisation@local_min=mace \
local_min.fmax=0.50 \
model=mace \
model.model_paths=models/MACE_SPICE2_NEUTRAL.model \
hydra.verbose=true
The following dictionaries are passed directly to the function of that name.
conformers(EmbedMultipleConfs)minimisation.MMFFGetMoleculePropertiesminimisation.MMFFGetMoleculeForceFieldenergy_eval.MMFFGetMoleculePropertiesenergy_eval.MMFFGetMoleculeForceField
The hydra config is set up to allow additional kwargs to be passed to these functions e.g. +minimisation.MMFFGetMoleculeProperties.mmffVerbosity=1.
Common kwargs
threshold(set by default to 16.1 kcal/mol - calibrated using LigBoundConf 2.0)conformers.numConfsglobal_min.maxIters/local_min.maxItersglobal_min.fmax/local_min.maxItershydra.verboseseed
strain-relief accepts pd.DataFrames with RDKit molecules stored as bytes strings (using mol.ToBinary())
Logging is set to the INFO level by default which logs only aggregate information. hydra.verbose=true can be used to activate DEBUG level logging which includes information for every molecule and conformer.
pytest tests/- runs all tests (unit and integration)pytest tests/ -m "not gpu"- excludes all MACE testspytest tests/ -m "not integration"- runs all unit tests
NB Tests requiring the eSEN model will be skipped if an eSEN model is not located in tests/models/eSEN.pt.
If you use StrainRelief or adapt the StrainRelief code for any purpose, please cite:
@misc{wallace2025strainrelief,
title={Strain Problems got you in a Twist? Try StrainRelief: A Quantum-Accurate Tool for Ligand Strain Calculations},
author={Ewan R. S. Wallace and Nathan C. Frey and Joshua A. Rackers},
year={2025},
eprint={2503.13352},
archivePrefix={arXiv},
primaryClass={physics.chem-ph},
url={https://arxiv.org/abs/2503.13352},
}@article{batatia2022mace,
title={MACE: Higher order equivariant message passing neural networks for fast and accurate force fields},
author={Batatia, Ilyes and Kovacs, David P and Simm, Gregor and Ortner, Christoph and Cs{\'a}nyi, G{\'a}bor},
journal={Advances in neural information processing systems},
volume={35},
pages={11423--11436},
year={2022}
}For any questions, please reach out to Ewan Wallace: [email protected]
