google / evojax

Apache License 2.0
846 stars 87 forks source link

Update for `evosax` API changes #36

Closed RobertTLange closed 2 years ago

RobertTLange commented 2 years ago

Updates the strategies, which rely on the evosax backend after some minor frontend changes in version 0.0.9. The evoJAX frontend stays untouched and the strategy functionalities internally have also not changed. But evosax internally now uses flax-based dataclasses for the state (EvoState) and hyperparameters (EvoParams), which make type-checking easier and have all the benefits of immutable states, etc.