Add link to code

README.md

@@ -24,9 +24,11 @@ EM = 80.85
 F1 = 87.94
 ```
 
-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},