Add link to code

README.md

@@ -37,9 +37,11 @@ The dev-set performance reported in the paper is averaged over four seeds, and w
 | stdev         | 0.061 | 0.079 |
 ```
 
-Code: _coming soon_
+Code: [https://github.com/neuralmagic/sparseml/tree/main/research/optimal_BERT_surgeon_oBERT](https://github.com/neuralmagic/sparseml/tree/main/research/optimal_BERT_surgeon_oBERT)
 
-## BibTeX entry and citation info
+If you find the model useful, please consider citing our work.
+
+## Citation info
 ```bibtex
 @article{kurtic2022optimal,
   title={The Optimal BERT Surgeon: Scalable and Accurate Second-Order Pruning for Large Language Models},