Add link to code

README.md

@@ -36,9 +36,11 @@ The dev-set performance reported in the paper is averaged over three seeds, and
 | stdev        | 0.115 | 0.175 |
 ```
 
-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},