search

bischtob / Opterax

An gradient-free optimization suite written in JAX. We conform to the optax interface and provide ensemble-based optimizers.
Other
1 stars 0 forks source link

Implement first iteration of optimizer class and tests #1

Closed bischtob closed 3 years ago

bischtob commented 3 years ago

We need to implement the first version of an optimizer class. Test need to be implemented to compare against existing codebases.

bischtob commented 3 years ago

For this, we should reuse the learnings from optax -> transformations. May make sense to build to a similar interface. Will post updates here.