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.7.tar.gz
(6.3 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.7-py3-none-any.whl
(6.7 kB
view details)
File details
Details for the file mutax-0.4.7.tar.gz.
File metadata
- Download URL: mutax-0.4.7.tar.gz
- Upload date:
- Size: 6.3 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 |
233767f90e90131ce7eda86b51c56ed864a301c660f91262b9a71f292532adf4
|
|
| MD5 |
c9ba0174319fe31f63841eb2f2aaf662
|
|
| BLAKE2b-256 |
870f476bfdcd660b7814e7806f50254001fb78c42c74a5501428679c97eac726
|
File details
Details for the file mutax-0.4.7-py3-none-any.whl.
File metadata
- Download URL: mutax-0.4.7-py3-none-any.whl
- Upload date:
- Size: 6.7 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 |
839b5dc5a898ff31baaeba00d9149f97c97720835048a561c18e9ee97763e1fe
|
|
| MD5 |
6bb0f978fcc08b16be8fed8dea20459a
|
|
| BLAKE2b-256 |
c858cf155bb549e7b663419b84f8d807c10ce0b9991f0add243c26d7ddd7c837
|
