Update README.md

README.md

@@ -46,4 +46,18 @@ https://github.com/luffycodes/Tutorbot-Spock-Bio
 Apache License 2.0
 
 **Where to send questions or comments about the model:**
-Shashank Sonkar (ss164@rice.edu)
+Shashank Sonkar (ss164@rice.edu)
+
+If you use this work, please cite:
+Investigating the Role of Feed-Forward Networks in Transformers Using Parallel Attention and Feed-Forward Net Design
+https://arxiv.org/abs/2305.13297
+```
+@misc{sonkar2023investigating,
+      title={Investigating the Role of Feed-Forward Networks in Transformers Using Parallel Attention and Feed-Forward Net Design}, 
+      author={Shashank Sonkar and Richard G. Baraniuk},
+      year={2023},
+      eprint={2305.13297},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```