{"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/tensornet-cartesian-tensor-representations","title":"TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials","arxiv_id":"2306.06482","date":"2023-06-10","proceeding":"NeurIPS 2023 11","authors":["Guillem Simeon","Gianni de Fabritiis"],"abstract":"The development of efficient machine learning models for molecular systems representation is becoming crucial in scientific research. We introduce TensorNet, an innovative O(3)-equivariant message-passing neural network architecture that leverages Cartesian tensor representations. By using Cartesian tensor atomic embeddings, feature mixing is simplified through matrix product operations. Furthermore, the cost-effective decomposition of these tensors into rotation group irreducible representations allows for the separate processing of scalars, vectors, and tensors when necessary. Compared to higher-rank spherical tensor models, TensorNet demonstrates state-of-the-art performance with significantly fewer parameters. For small molecule potential energies, this can be achieved even with a single interaction layer. As a result of all these properties, the model's computational cost is substantially decreased. Moreover, the accurate prediction of vector and tensor molecular quantities on top of potential energies and forces is possible. In summary, TensorNet's framework opens up a new space for the design of state-of-the-art equivariant models.","url_abs":"https://arxiv.org/abs/2306.06482v2","url_pdf":"https://arxiv.org/pdf/2306.06482v2.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":"tensornet-cartesian-tensor-representations","repo_url":"https://github.com/torchmd/torchmd-net","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":null},{"paper_slug":"tensornet-cartesian-tensor-representations","repo_url":"https://github.com/materialsvirtuallab/matgl","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"BSD-3-Clause"}},{"paper_slug":"tensornet-cartesian-tensor-representations","repo_url":"https://github.com/jla-gardner/graph-pes","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":0,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"formation-energy","task_name":"Formation Energy"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[{"leaderboard":"/sota/formation-energy-on-qm9","task":"Formation Energy","dataset":"QM9","model":"TensorNet","rank_in_archive_order":1,"of":18,"metrics":{"MAE":"0.09"},"uses_additional_data":false}],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2306.06482","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2306.06482"}},"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":"deterministic:regex_extraction","url":"https://github.com/torchmd/torchmd-net","reach":null},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/materialsvirtuallab/matgl","reach":{"status":"ok","spdx":"BSD-3-Clause"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/jla-gardner/graph-pes","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran_draft_wrong":2,"ran_fixture":1,"unverified":4},"by_repo_kind":{"official":{"samples":3,"ran":3,"repositories":1},"listed":{"samples":4,"ran":0,"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":"28535c73918bb5b8","entry":"I_to_tensor","repo":"torchmd/torchmd-net","repo_kind":"official","path":"torchmdnet/models/tensornet.py","file_url":"https://github.com/torchmd/torchmd-net/blob/HEAD/torchmdnet/models/tensornet.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"28535c73918bb5b8"}},{"code_sha256_prefix":"a742180f11e48f3f","entry":"skewtensor_to_vector","repo":"torchmd/torchmd-net","repo_kind":"official","path":"torchmdnet/models/tensornet.py","file_url":"https://github.com/torchmd/torchmd-net/blob/HEAD/torchmdnet/models/tensornet.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a742180f11e48f3f"}},{"code_sha256_prefix":"8094daa092ae20f8","entry":"vector_to_skewtensor","repo":"torchmd/torchmd-net","repo_kind":"official","path":"torchmdnet/models/tensornet.py","file_url":"https://github.com/torchmd/torchmd-net/blob/HEAD/torchmdnet/models/tensornet.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"8094daa092ae20f8"}},{"code_sha256_prefix":"7b2030d7996e1a92","entry":"compute_pair_vector_and_distance","repo":"materialsvirtuallab/matgl","repo_kind":"listed","path":"src/matgl/graph/_compute.py","file_url":"https://github.com/materialsvirtuallab/matgl/blob/HEAD/src/matgl/graph/_compute.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"BSD-3-Clause","inline_ok":true,"mcp_get_code":{"code_sha256":"7b2030d7996e1a92"}},{"code_sha256_prefix":"e182b3ac08bf7ffd","entry":"compute_theta_and_phi","repo":"materialsvirtuallab/matgl","repo_kind":"listed","path":"src/matgl/graph/_compute.py","file_url":"https://github.com/materialsvirtuallab/matgl/blob/HEAD/src/matgl/graph/_compute.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"BSD-3-Clause","inline_ok":true,"mcp_get_code":{"code_sha256":"e182b3ac08bf7ffd"}},{"code_sha256_prefix":"6901492b3bc4dc4f","entry":"ensure_batch_attribute","repo":"materialsvirtuallab/matgl","repo_kind":"listed","path":"src/matgl/graph/data.py","file_url":"https://github.com/materialsvirtuallab/matgl/blob/HEAD/src/matgl/graph/data.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"BSD-3-Clause","inline_ok":true,"mcp_get_code":{"code_sha256":"6901492b3bc4dc4f"}},{"code_sha256_prefix":"f8f4216e12bc9e70","entry":"split_dataset","repo":"materialsvirtuallab/matgl","repo_kind":"listed","path":"src/matgl/graph/data.py","file_url":"https://github.com/materialsvirtuallab/matgl/blob/HEAD/src/matgl/graph/data.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"BSD-3-Clause","inline_ok":true,"mcp_get_code":{"code_sha256":"f8f4216e12bc9e70"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}