[Doc] Update citation (pytorch#1228)

ai4co · Jun 2, 2023 · 56d905b · 56d905b
1 parent 2485344
commit 56d905b
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/README.md b/README.md
@@ -13,8 +13,8 @@
 # TorchRL
 
 [**Documentation**](#documentation-and-knowledge-base) | [**TensorDict**](#writing-simplified-and-portable-rl-codebase-with-tensordict) |
-[**Features**](#features) | [**Examples, tutorials and demos**](#examples-tutorials-and-demos) | [**Installation**](#installation) |
-[**Asking a question**](#asking-a-question) | [**Citation**](#citation) | [**Contributing**](#contributing)
+[**Features**](#features) | [**Examples, tutorials and demos**](#examples-tutorials-and-demos) | [**Citation**](#citation) | [**Installation**](#installation) |
+[**Asking a question**](#asking-a-question) | [**Contributing**](#contributing)
 
 **TorchRL** is an open-source Reinforcement Learning (RL) library for PyTorch.
 
@@ -504,7 +504,22 @@ about handling the various configuration settings.
 We also provide [tutorials and demos](https://pytorch.org/rl/#tutorials) that give a sense of
 what the library can do.
 
+## Citation
+
+If you're using TorchRL, please refer to this BibTeX entry to cite this work:
+```
+@misc{bou2023torchrl,
+      title={TorchRL: A data-driven decision-making library for PyTorch}, 
+      author={Albert Bou and Matteo Bettini and Sebastian Dittert and Vikash Kumar and Shagun Sodhani and Xiaomeng Yang and Gianni De Fabritiis and Vincent Moens},
+      year={2023},
+      eprint={2306.00577},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG}
+}
+```
+
 ## Installation
+
 Create a conda environment where the packages will be installed.
 
 ```
@@ -635,19 +650,6 @@ If you spot a bug in the library, please raise an issue in this repo.
 If you have a more generic question regarding RL in PyTorch, post it on
 the [PyTorch forum](https://discuss.pytorch.org/c/reinforcement-learning/6).
 
-## Citation
-
-If you're using TorchRL, please refer to this BibTeX entry to cite this work:
-```
-@software{TorchRL,
-  author = {Moens, Vincent},
-  title = {{TorchRL: an open-source Reinforcement Learning (RL) library for PyTorch}},
-  url = {https://github.com/pytorch/rl},
-  version = {0.1.1},
-  year = {2023}
-}
-```
-
 ## Contributing
 
 Internal collaborations to torchrl are welcome! Feel free to fork, submit issues and PRs.