Merge pull request #70 from EMI-Group/dev_es

Complete paper annotations for all es algorithms.

src/evox/algorithms/so/es/nes.py

@@ -1,3 +1,13 @@
+# --------------------------------------------------------------------------------------
+# This code implements algorithms described in the following papers:
+#
+# Title: Exponential Natural Evolution Strategies (xNES)
+# Link: https://dl.acm.org/doi/abs/10.1145/1830483.1830557
+#
+# Title: Natural Evolution Strategies (SeparableNES)
+# Link: https://www.jmlr.org/papers/volume15/wierstra14a/wierstra14a.pdf
+# --------------------------------------------------------------------------------------
+
 import math
 import jax
 import jax.numpy as jnp

src/evox/algorithms/so/es/open_es.py

@@ -1,3 +1,13 @@
+# --------------------------------------------------------------------------------------
+# This code implements algorithms described in the following papers:
+#
+# Title: Exponential Natural Evolution Strategies (xNES)
+# Link: https://dl.acm.org/doi/abs/10.1145/1830483.1830557
+#
+# Title: Natural Evolution Strategies (SeparableNES)
+# Link: https://www.jmlr.org/papers/volume15/wierstra14a/wierstra14a.pdf
+# --------------------------------------------------------------------------------------
+
 import jax
 import jax.numpy as jnp
 import optax

src/evox/algorithms/so/es/pgpe.py

@@ -1,3 +1,9 @@
+# --------------------------------------------------------------------------------------
+# 1. This code implements the algorithm described in the following paper:
+# Title: Parameter-exploring Policy Gradients (PGPE)
+# Link: https://mediatum.ub.tum.de/doc/1287490/file.pdf
+# --------------------------------------------------------------------------------------
+
 import jax
 import jax.numpy as jnp
 import optax