SciPy-like differential evolution for JAX
Fully jitted optimization of any JAX-compatible function. Serial and parallel execution on CPU, GPU, and TPU.
Installation
pip install mutax
Quick start
import jax.numpy as jnp
from mutax import differential_evolution
def cost_function(xs):
return jnp.sum(xs**2)
bounds = [(-5, 5)] * 10 # 10-dimensional problem with bounds for each dimension
result = differential_evolution(cost_function, bounds)
print("Best solution:", result.x)
print("Objective value:", result.fun)
Documentation
The documentation is available at Read the Docs.
1 maintainer
Metadata
1 maintainer
Release files for mutax 0.5.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| mutax-0.5.1.tar.gz | 6.5 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| mutax-0.5.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 13.4 kB
Release files / mutax-0.5.1.tar.gz
| Download URL | mutax-0.5.1.tar.gz |
|---|---|
| Size | 6.5 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
03abf31b3a76b8ae88ca3f66d250aa43bb2220c214d02af71f57ad3e4e6a82c5
|
|
BLAKE2b-256 checksum How to use checksums |
a1302bc324776528d229c12d9d7e11534c258c5b8b14449dff58ff4f1ab841a6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.12.7 {"installer":{"name":"uv","version":"0.12.7","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
Release files / mutax-0.5.1-py3-none-any.whl
| Download URL | mutax-0.5.1-py3-none-any.whl |
|---|---|
| Size | 6.9 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
9aa4961123ba5e6a7cb04377942e07f559df4e8523d80899a78dfbaf4b2c69b9
|
|
BLAKE2b-256 checksum How to use checksums |
b2505ebc727e29cb87cb3352aeafeba70f043d313d9ac5f0c6221b5a5ce2b0d7
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.12.7 {"installer":{"name":"uv","version":"0.12.7","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
