feat: README.md

Author: gerdm

README.md

@@ -3,8 +3,13 @@
 ![intro-plot](https://github.com/gerdm/weighted-likelihood-filter/assets/4108759/f32c0b01-433b-46e6-9153-262e1b6c4f10)
 
 ## Installation
-To run the experiments, make sure to have installed `jax>=0.4.2`, [rebayes-mini](https://github.com/gerdm/rebayes-mini/tree/main), and the [BayesianOptimization](https://github.com/bayesian-optimization/BayesianOptimization) package:
+To run the experiments, make sure to have installed `jax>=0.4.2`,
+[rebayes-mini](https://github.com/gerdm/rebayes-mini/tree/main),
+[flax](https://github.com/google/flax),
+and the [BayesianOptimization](https://github.com/bayesian-optimization/BayesianOptimization) package:
+
 ```bash
 pip install git+https://github.com/gerdm/rebayes-mini.git
+pip install flax
 pip install bayesian-optimization
 ```