diff --git a/LICENSE b/LICENSE
new file mode 100644
index 0000000..6842536
--- /dev/null
+++ b/LICENSE
@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 InstaDeep Ltd
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
diff --git a/README.md b/README.md
index c7f3e89..c6d7af4 100644
--- a/README.md
+++ b/README.md
@@ -251,4 +251,19 @@ The public datasets included in this repository originate from the following pub
 Please credit the original authors when using these data in derivative works or publications.
 
 ## Citation
-If you use this codebase in your research, please cite the associated paper once available. A BibTeX entry will be added here upon publication.
+If you use this codebase in your research, please cite the associated paper:
+
+```bibtex
+@misc{amiaudplachy2026bayespdexploringsequencebinding,
+      title={Bayes-PD: Exploring a Sequence to Binding Bayesian Neural Network model trained on Phage Display data},
+      author={Ilann Amiaud-Plachy and Michael Blank and Oliver Bent and Sebastien Boyer},
+      year={2026},
+      eprint={2601.03930},
+      archivePrefix={arXiv},
+      primaryClass={q-bio.PE},
+      url={https://arxiv.org/abs/2601.03930},
+}
+```
+
+## License
+This project is released under the [MIT License](LICENSE).