Predicting heavy-metal retention in montmorillonite: Physics-informed machine learning guided by XRD hydration states and molecular dynamics | Synapse