evorl.algorithms.ec.mo.nsga2_brax

Module Contents

Classes

NSGA2Workflow

API

class evorl.algorithms.ec.mo.nsga2_brax.NSGA2Workflow(*, env: evorl.envs.Env, agent: evorl.agent.Agent, ec_optimizer: evorl.ec.optimizers.EvoOptimizer, ec_evaluator: evorl.evaluators.Evaluator | evorl.evaluators.EpisodeCollector, agent_state_vmap_axes: evorl.agent.AgentStateAxis = 0, config: omegaconf.DictConfig)

Bases: evorl.workflows.MultiObjectiveECWorkflowTemplate

learn(state: evorl.types.State) → evorl.types.State

classmethod name()