evox 0.0.2

evox

EvoX: A Distributed GPU-accelerated Library for Evolutionary Computation

[Docs] [Paper]

Features

• Single-objective and multi-objective algorithms.
• GPU computing.
• Easy to use distributed pipeline.
• Support a wide range of problems.

Index

• Getting started
• Example

Installation

pip install evox

Getting started

To start with, import evox

import evox
from evox import algorithm, problem, pipeline

Then, create an algorithm and a problem:

pso = algorithms.PSO(
    lb=jnp.full(shape=(2,), fill_value=-32),
    ub=jnp.full(shape=(2,), fill_value=32),
    pop_size=100,
)
ackley = problems.classic.Ackley()

The algorithm and the problem are composed together using pipeline:

pipeline = pipelines.StdPipeline(pso, ackley)

To initialize the whole pipeline, call init on the pipeline object with a PRNGKey. Calling init will recursively initialize a tree of objects, meaning the algorithm pso and problem ackley are automatically initialize as well.

key = jax.random.PRNGKey(42)
state = pipeline.init(key)

To run the pipeline, call step on the pipeline.

# run the pipeline for 100 steps
for i in range(100):
    state = pipeline.step(state)

For more detailed usage, please refer to our documentation.

Exmaple

The example folder has many examples on how to use EvoX.