Update README.md

Author: nahid18

README.md

@@ -14,10 +14,18 @@ using only matrix-vector products. ``traceax`` is heavily inspired by
 [lineax](https://github.com/patrick-kidger/lineax) as well as
 [XTrace](https://github.com/eepperly/XTrace).
 
+> [!NOTE]
+> For preprint, please see: <br/>
+> Nahid, A.A., Serafin, L., Mancuso, N. (2025). <i>traceax</i>: a JAX-based framework for stochastic trace estimation. bioRxiv 
+> (https://doi.org/10.1101/2025.07.14.662216)
+
+> [!IMPORTANT]
+> To reproduce the analyses in the preprint, please see: <br/>
+> https://github.com/mancusolab/traceax_analysis
+
   [**Installation**](#installation)
   | [**Example**](#get-started-with-example)
   | [**Documentation**](#documentation)
-  | [**Citation**](#citation)
   | [**Notes**](#notes)
   | [**Support**](#support)
   | [**Other Software**](#other-software)
@@ -94,11 +102,6 @@ print(nt.estimate(key4, operator, k))  # (Array(3.3297246, dtype=float32), {'std
 ## Documentation
 Documentation is available at [here](https://mancusolab.github.io/traceax/).
 
-## Citation
-If you use `traceax` in your work, please cite:
-
-> Nahid, A.A., Serafin, L., Mancuso, N. (2025). <i>traceax</i>: a JAX-based framework for stochastic trace estimation. bioRxiv (https://doi.org/10.1101/2025.07.14.662216)
-
 ## Notes
 
 -   `traceax` uses [JAX](https://github.com/google/jax) with [Just In