RobertTLange / evosax

Evolution Strategies in JAX 🦎
Apache License 2.0
475 stars 44 forks source link

Incorporate experimental decoding features into the package #58

Closed kmheckel closed 5 months ago

kmheckel commented 1 year ago

Hi Rob, (@RobertTLange )

I've used evosax for a number of school projects and found it great! I'm currently building a Spiking Neural Network library in JAX and was hoping to leverage the hypernetwork/random projection decoding methods but they're currently not incorporated into the PyPI release of evosax. Is there a pathway to them being included into the package as a regular feature?

Thanks! Kade

kmheckel commented 1 year ago

For what it's worth, I've created a small library/package built on top of Deepmind-Haiku that implements random projection, discrete cosine transform from compressed weight search, and a hypernetwork.

I'm currently using evosax to evaluate neuroevolution for SNNs and trying to figure out how to extract high performance from them - the dynamics of SNNs seem to be posing quite the challenge for evolutionary approaches so far.

Synecdoche

Thanks for your work on evosax!