Update README.md (#294)

* Update README.md

* Update index.rst

Author: guillemsimeon

README.md

@@ -86,6 +86,31 @@ CUDA_VISIBLE_DEVICES=0,1 torchmd-train --conf torchmd-net/examples/ET-QM9.yaml.y
 If you use TorchMD-NET in your research, please cite the following papers:
 
 #### Main reference
+
+```
+@misc{pelaez2024torchmdnet,
+title={TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulations}, 
+author={Raul P. Pelaez and Guillem Simeon and Raimondas Galvelis and Antonio Mirarchi and Peter Eastman and Stefan Doerr and Philipp Thölke and Thomas E. Markland and Gianni De Fabritiis},
+year={2024},
+eprint={2402.17660},
+archivePrefix={arXiv},
+primaryClass={cs.LG}
+}
+```
+
+#### TensorNet
+
+```
+@inproceedings{simeon2023tensornet,
+title={TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials},
+author={Guillem Simeon and Gianni De Fabritiis},
+booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
+year={2023},
+url={https://openreview.net/forum?id=BEHlPdBZ2e}
+}
+```
+
+#### Equivariant Transformer
 ```
 @inproceedings{
 tholke2021equivariant,
@@ -115,18 +140,6 @@ url={https://openreview.net/forum?id=zNHzqZ9wrRB}
 }
 ```
 
-#### TensorNet
-
-```
-@inproceedings{simeon2023tensornet,
-title={TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials},
-author={Guillem Simeon and Gianni De Fabritiis},
-booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
-year={2023},
-url={https://openreview.net/forum?id=BEHlPdBZ2e}
-}
-```
-
 ## Developer guide
 
 ### Implementing a new architecture

docs/source/index.rst

@@ -10,6 +10,36 @@ Cite
 If you use TorchMD-NET in your research, please cite the following papers:
 
 Main reference
+~~~~~~~~~
+
+.. code-block:: bibtex
+
+    @misc{
+    pelaez2024torchmdnet,
+    title={TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulations},
+    author={Raul P. Pelaez and Guillem Simeon and Raimondas Galvelis and Antonio Mirarchi and Peter Eastman and Stefan Doerr and Philipp Thölke and Thomas E. Markland and Gianni De Fabritiis},
+    year={2024},
+    eprint={2402.17660},
+    archivePrefix={arXiv},
+    primaryClass={cs.LG}
+    }
+
+TensorNet
+~~~~~~~~~
+
+.. code-block:: bibtex
+
+    @inproceedings{
+    simeon2023tensornet,
+    title={TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials},
+    author={Guillem Simeon and Gianni De Fabritiis},
+    booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
+    year={2023},
+    url={https://openreview.net/forum?id=BEHlPdBZ2e}
+    }
+
+
+Equivariant Transformer
 ~~~~~~~~~~~~~~
 
 .. code-block:: bibtex
@@ -43,20 +73,6 @@ Graph Network
   month = sep 
   }
 
-TensorNet
-~~~~~~~~~
-
-.. code-block:: bibtex
-
-    @inproceedings{
-    simeon2023tensornet,
-    title={TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials},
-    author={Guillem Simeon and Gianni De Fabritiis},
-    booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
-    year={2023},
-    url={https://openreview.net/forum?id=BEHlPdBZ2e}
-    }
-
 
 .. toctree::
    :hidden: