Update README.md and main.md

README.md

@@ -6,9 +6,16 @@
   <img src="https://img.alicdn.com/imgextra/i1/O1CN01lvLpfw25Pl4ohGZnU_!!6000000007519-2-tps-1628-490.png" alt="Trinity-RFT" style="height: 120px;">
 </div>
 
-
 &nbsp;
 
+<div align="center">
+
+[![paper](http://img.shields.io/badge/cs.LG-2505.17826-B31B1B?logo=arxiv&logoColor=red)](https://arxiv.org/abs/2505.17826)
+[![doc](https://img.shields.io/badge/Docs-blue?logo=markdown)](https://modelscope.github.io/Trinity-RFT/)
+[![pypi](https://img.shields.io/pypi/v/trinity-rft?logo=pypi&color=026cad)](https://pypi.org/project/trinity-rft/0.1.0/)
+![license](https://img.shields.io/badge/license-Apache--2.0-000000.svg)
+
+</div>
 
 
 **Trinity-RFT is a general-purpose, flexible, scalable and user-friendly framework designed for reinforcement fine-tuning (RFT) of large language models (LLM).**
@@ -146,6 +153,12 @@ pip install flash-attn -v
 # pip install flash-attn -v --no-build-isolation
 ```
 
+Installation using pip:
+
+```shell
+pip install trinity-rft==0.1.0
+```
+
 Installation from docker:
 we have provided a dockerfile for Trinity-RFT (trinity)
 
@@ -332,10 +345,13 @@ This project is built upon many excellent open-source projects, including:
 
 ## Citation
 ```plain
-@misc{Trinity-RFT,
-  title={Trinity-RFT},
-  author={{Trinity-RFT Team}},
-  url={https://github.com/modelscope/trinity-rft},
-  year={2025},
+@misc{trinity-rft,
+      title={Trinity-RFT: A General-Purpose and Unified Framework for Reinforcement Fine-Tuning of Large Language Models},
+      author={Xuchen Pan and Yanxi Chen and Yushuo Chen and Yuchang Sun and Daoyuan Chen and Wenhao Zhang and Yuexiang Xie and Yilun Huang and Yilei Zhang and Dawei Gao and Yaliang Li and Bolin Ding and Jingren Zhou},
+      year={2025},
+      eprint={2505.17826},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2505.17826},
 }
 ```

docs/sphinx_doc/source/main.md

@@ -124,6 +124,11 @@ pip install flash-attn -v
 # pip install flash-attn -v --no-build-isolation
 ```
 
+Installation using pip:
+
+```shell
+pip install trinity-rft==0.1.0
+```
 
 Installation from docker:
 we have provided a dockerfile for Trinity-RFT (trinity)
@@ -319,10 +324,13 @@ This project is built upon many excellent open-source projects, including:
 
 ## Citation
 ```
-@misc{Trinity-RFT,
-  title={Trinity-RFT},
-  author={{Trinity-RFT Team}},
-  url={https://github.com/modelscope/trinity-rft},
-  year={2025},
+@misc{trinity-rft,
+      title={Trinity-RFT: A General-Purpose and Unified Framework for Reinforcement Fine-Tuning of Large Language Models},
+      author={Xuchen Pan and Yanxi Chen and Yushuo Chen and Yuchang Sun and Daoyuan Chen and Wenhao Zhang and Yuexiang Xie and Yilun Huang and Yilei Zhang and Dawei Gao and Yaliang Li and Bolin Ding and Jingren Zhou},
+      year={2025},
+      eprint={2505.17826},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2505.17826},
 }
 ```