{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/equivariant-message-passing-for-the","title":"Equivariant message passing for the prediction of tensorial properties and molecular spectra","arxiv_id":"2102.03150","date":"2021-02-05","proceeding":null,"authors":["Kristof T. Schütt","Oliver T. Unke","Michael Gastegger"],"abstract":"Message passing neural networks have become a method of choice for learning on graphs, in particular the prediction of chemical properties and the acceleration of molecular dynamics studies. While they readily scale to large training data sets, previous approaches have proven to be less data efficient than kernel methods. We identify limitations of invariant representations as a major reason and extend the message passing formulation to rotationally equivariant representations. On this basis, we propose the polarizable atom interaction neural network (PaiNN) and improve on common molecule benchmarks over previous networks, while reducing model size and inference time. We leverage the equivariant atomwise representations obtained by PaiNN for the prediction of tensorial properties. Finally, we apply this to the simulation of molecular spectra, achieving speedups of 4-5 orders of magnitude compared to the electronic structure reference.","url_abs":"https://arxiv.org/abs/2102.03150v4","url_pdf":"https://arxiv.org/pdf/2102.03150v4.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"equivariant-message-passing-for-the","repo_url":"https://github.com/learningmatter-mit/NeuralForceField","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"pytorch","reach":null},{"paper_slug":"equivariant-message-passing-for-the","repo_url":"https://github.com/gerkone/painn-jax","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":0,"framework":"jax","reach":{"status":"unanswered"}}],"tasks":[{"task_slug":"drug-discovery","task_name":"Drug Discovery"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[{"leaderboard":"/sota/drug-discovery-on-qm9","task":"Drug Discovery","dataset":"QM9","model":"PaiNN","rank_in_archive_order":6,"of":11,"metrics":{"Error ratio":"0.411"},"uses_additional_data":false}],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2102.03150","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2102.03150"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/learningmatter-mit/NeuralForceField","reach":null},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/gerkone/painn-jax","reach":{"status":"unanswered"}}],"summary":{"ran":15,"ran_draft_wrong":5,"ran_honours":1,"unverified":12},"by_repo_kind":{"listed":{"samples":33,"ran":21,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"8ac6bce450fc85e3","entry":"CosineEnvelope","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"8ac6bce450fc85e3"}},{"code_sha256_prefix":"dc349a5755ad2883","entry":"Dense","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"dc349a5755ad2883"}},{"code_sha256_prefix":"6af9e88a6c54be01","entry":"DistanceEmbed","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"6af9e88a6c54be01"}},{"code_sha256_prefix":"952ca748b3f1516e","entry":"EmbeddingBlock","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"952ca748b3f1516e"}},{"code_sha256_prefix":"7f75d01bd99cc6dd","entry":"InvariantDense","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"7f75d01bd99cc6dd"}},{"code_sha256_prefix":"0d21a7509e419d0b","entry":"LearnableSwish","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"0d21a7509e419d0b"}},{"code_sha256_prefix":"b65822c621376024","entry":"PainnRadialBasis","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"b65822c621376024"}},{"code_sha256_prefix":"802d844e902371ef","entry":"ScaleShift","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"802d844e902371ef"}},{"code_sha256_prefix":"7f836403bedf830d","entry":"Swish","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"7f836403bedf830d"}},{"code_sha256_prefix":"5cfab56ad64af6fe","entry":"add_embedding","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"5cfab56ad64af6fe"}},{"code_sha256_prefix":"2eece0db59be49c0","entry":"att_readout_probs","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"2eece0db59be49c0"}},{"code_sha256_prefix":"2ac65d5ac1bf4b38","entry":"compute_grad","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"2ac65d5ac1bf4b38"}},{"code_sha256_prefix":"07fc732b54bba7cd","entry":"gen","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"07fc732b54bba7cd"}},{"code_sha256_prefix":"18fd70ccb86319a7","entry":"get_offsets","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"18fd70ccb86319a7"}},{"code_sha256_prefix":"2cfb01eec4fba085","entry":"get_rij","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"2cfb01eec4fba085"}},{"code_sha256_prefix":"3e1aa28156810aba","entry":"make_directed","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"3e1aa28156810aba"}},{"code_sha256_prefix":"3df02af32c33da5a","entry":"norm","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"3df02af32c33da5a"}},{"code_sha256_prefix":"6565f4a40973667d","entry":"preprocess_r","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"6565f4a40973667d"}},{"code_sha256_prefix":"04e4a832f331db07","entry":"scatter_add","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"04e4a832f331db07"}},{"code_sha256_prefix":"42c43fca053545ea","entry":"shifted_softplus","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"42c43fca053545ea"}},{"code_sha256_prefix":"a929e9679ef4134f","entry":"to_module","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a929e9679ef4134f"}},{"code_sha256_prefix":"55d8c70e377b4a86","entry":"AttentionPool","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"55d8c70e377b4a86"}},{"code_sha256_prefix":"f09c070242efedbf","entry":"InvariantMessage","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f09c070242efedbf"}},{"code_sha256_prefix":"670901a6573e1934","entry":"MeanPool","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"670901a6573e1934"}},{"code_sha256_prefix":"573a7b800820d725","entry":"MessageBase","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"573a7b800820d725"}},{"code_sha256_prefix":"9f1980fc08094e16","entry":"MessageBlock","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"9f1980fc08094e16"}},{"code_sha256_prefix":"398bfbf43b702ff8","entry":"MolFpPool","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"398bfbf43b702ff8"}},{"code_sha256_prefix":"450568a63ba4bedb","entry":"Painn","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"450568a63ba4bedb"}},{"code_sha256_prefix":"05495644c9588ed2","entry":"ReadoutBlock","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"05495644c9588ed2"}},{"code_sha256_prefix":"4b3ac9cd9d28db18","entry":"SumPool","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"4b3ac9cd9d28db18"}},{"code_sha256_prefix":"1418df20e13d6afe","entry":"UpdateBlock","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"1418df20e13d6afe"}},{"code_sha256_prefix":"d31febe9a5aa1b51","entry":"add_stress","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"d31febe9a5aa1b51"}},{"code_sha256_prefix":"5d9228f747be613d","entry":"sum_and_grad","repo":"learningmatter-mit/NeuralForceField","repo_kind":"listed","path":"nff/nn/models/painn.py","file_url":"https://github.com/learningmatter-mit/NeuralForceField/blob/HEAD/nff/nn/models/painn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"5d9228f747be613d"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}