Skip to content

Commit 60cd38b

Browse files
committed
Fix link notebook ME-LS + show example how to jit MAP-Elites update function
1 parent e8ce2f0 commit 60cd38b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

README.md

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,10 +114,13 @@ centroids = compute_euclidean_centroids(
114114
key, subkey = jax.random.split(key)
115115
repertoire, emitter_state, metrics = map_elites.init(init_variables, centroids, subkey)
116116

117+
# Jit the update function for faster iterations
118+
update_fn = jax.jit(map_elites.update)
119+
117120
# Run MAP-Elites loop
118121
for i in range(num_iterations):
119122
key, subkey = jax.random.split(key)
120-
(repertoire, emitter_state, metrics,) = map_elites.update(
123+
(repertoire, emitter_state, metrics,) = update_fn(
121124
repertoire,
122125
emitter_state,
123126
subkey,
@@ -146,7 +149,7 @@ QDax currently supports the following algorithms:
146149
| [Multi-Objective MAP-Elites (MOME)](https://arxiv.org/abs/2202.03057) | [![Open All Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/adaptive-intelligent-robotics/QDax/blob/main/examples/mome.ipynb) |
147150
| [MAP-Elites Evolution Strategies (MEES)](https://dl.acm.org/doi/pdf/10.1145/3377930.3390217) | [![Open All Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/adaptive-intelligent-robotics/QDax/blob/main/examples/mees.ipynb) |
148151
| [MAP-Elites PBT (ME-PBT)](https://openreview.net/forum?id=CBfYffLqWqb) | [![Open All Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/adaptive-intelligent-robotics/QDax/blob/main/examples/me_sac_pbt.ipynb) |
149-
| [MAP-Elites Low-Spread (ME-LS)](https://dl.acm.org/doi/abs/10.1145/3583131.3590433) | [![Open All Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/adaptive-intelligent-robotics/QDax/blob/main/examples/me_ls.ipynb) |
152+
| [MAP-Elites Low-Spread (ME-LS)](https://dl.acm.org/doi/abs/10.1145/3583131.3590433) | [![Open All Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/adaptive-intelligent-robotics/QDax/blob/main/examples/mels.ipynb) |
150153

151154

152155
## QDax baseline algorithms

0 commit comments

Comments
 (0)