diff --git a/README.md b/README.md index dfed34b..7278733 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -Hyperoptax Logo +Hyperoptax Logo # Hyperoptax: Parallel hyperparameter tuning with JAX @@ -29,10 +29,13 @@ If you do not yet have JAX installed, pick the right wheel for your accelerator: pip install --upgrade "jax[cpu]" # or GPU/TPU – see the official JAX installation guide ``` + ## 🥜 In a nutshell + All optimizers follow the same stateless pattern: `Optimizer.init` returns a `(state, optimizer)` pair, and `optimizer.optimize` runs the search loop. Your objective function must have the signature `fn(key, params) -> scalar`. Importantly, `params` can be _any_ PyTree. + ```python import jax from hyperoptax import BayesianSearch, LogSpace, LinearSpace diff --git a/assets/logo-transparent.png b/assets/logo-transparent.png index 1192243..4619fcb 100644 Binary files a/assets/logo-transparent.png and b/assets/logo-transparent.png differ