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
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mutax-0.4.8.tar.gz
(6.4 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
mutax-0.4.8-py3-none-any.whl
(6.8 kB
view details)
File details
Details for the file mutax-0.4.8.tar.gz.
File metadata
- Download URL: mutax-0.4.8.tar.gz
- Upload date:
- Size: 6.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
uv/0.12.1 {"installer":{"name":"uv","version":"0.12.1","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}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
73d6a3ea943c9d34cce0949e83a0725122d6b34830eb1ba44c3605e4d0f30480
|
|
| MD5 |
dbda6222f8734c1443e50cc65dc0ab25
|
|
| BLAKE2b-256 |
4651310254736241fbdec6e660f186fae2c321a65a3e8fc386f76e809ad0da48
|
File details
Details for the file mutax-0.4.8-py3-none-any.whl.
File metadata
- Download URL: mutax-0.4.8-py3-none-any.whl
- Upload date:
- Size: 6.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
uv/0.12.1 {"installer":{"name":"uv","version":"0.12.1","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}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ad2dd9e24759ff744306eca042ef32993c095fa536d45bf79e9e40ea3c100356
|
|
| MD5 |
3140ba6a3c0ffa082657f1af6423929d
|
|
| BLAKE2b-256 |
0166b37021d802258a36fdd1e5458d128f895d627f6753f20abb64cf72ef4414
|
