From 56d905bb8549aca12d99040f28c9fa6610927117 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Fri, 2 Jun 2023 17:23:34 +0100
Subject: [PATCH] [Doc] Update citation (#1228)

---
 README.md | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/README.md b/README.md
index 3c860680571..140c2fe6624 100644
--- 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.