Update README.md

README.md

@@ -69,5 +69,22 @@ action = model.step(obs, goal)
 - **Learning Rate**: 0.0001
 - **Weight Decay**: 0.05
 
+
+## Citation
+
+If you found the code usefull, please cite our work:
+
+```bibtex
+@misc{reuss2024efficient,
+    title={Efficient Diffusion Transformer Policies with Mixture of Expert Denoisers for Multitask Learning},
+    author={Moritz Reuss and Jyothish Pari and Pulkit Agrawal and Rudolf Lioutikov},
+    year={2024},
+    eprint={2412.12953},
+    archivePrefix={arXiv},
+    primaryClass={cs.LG}
+}
+```
+
+
 ## License
 This model is released under the MIT license.