https://github.com/gerlero/mutax

🎯 SciPy-like differential evolution optimization in JAX
https://github.com/gerlero/mutax

differential-evolution evolutionary-algorithms jax optimization optimization-algorithms optimization-methods typed

Last synced: 3 months ago
JSON representation

🎯 SciPy-like differential evolution optimization in JAX

• Host: GitHub
• URL: https://github.com/gerlero/mutax
• Owner: gerlero
• License: apache-2.0
• Created: 2025-09-14T13:24:47.000Z (4 months ago)
• Default Branch: main
• Last Pushed: 2025-10-02T13:49:16.000Z (3 months ago)
• Last Synced: 2025-10-02T15:26:07.580Z (3 months ago)
• Topics: differential-evolution, evolutionary-algorithms, jax, optimization, optimization-algorithms, optimization-methods, typed
• Language: Python
• Homepage: https://mutax.readthedocs.io
• Size: 252 KB
• Stars: 3
• Watchers: 0
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.txt

Awesome Lists containing this project

README

          


  


  **[SciPy-like](https://docs.scipy.org/doc/scipy/reference/generated/scipy.optimize.differential_evolution.html) differential evolution for [JAX](https://github.com/jax-ml/jax)**

  Fully [jitted](https://docs.jax.dev/en/latest/_autosummary/jax.jit.html#jax.jit) optimization of any JAX-compatible function. Serial and parallel execution on CPU, GPU, and TPU.

  [![Documentation](https://img.shields.io/readthedocs/mutax)](https://mutax.readthedocs.io/)

  [![CI](https://github.com/gerlero/mutax/actions/workflows/ci.yml/badge.svg)](https://github.com/gerlero/mutax/actions/workflows/ci.yml)

  [![Codecov](https://codecov.io/gh/gerlero/mutax/branch/main/graph/badge.svg)](https://codecov.io/gh/gerlero/mutax)

  [![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)

  [![ty](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ty/main/assets/badge/v0.json)](https://github.com/astral-sh/ty)

  [![uv](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json)](https://github.com/astral-sh/uv)

  [![Publish](https://github.com/gerlero/mutax/actions/workflows/pypi-publish.yml/badge.svg)](https://github.com/gerlero/mutax/actions/workflows/pypi-publish.yml)

  [![PyPI](https://img.shields.io/pypi/v/mutax)](https://pypi.org/project/mutax/)

  [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mutax)](https://pypi.org/project/mutax/)



## Installation

```bash

pip install mutax

```

## Quick start

```python

import jax.numpy as jnp

from mutax import differential_evolution

def cost_function(x):

    return jnp.sum(x**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](https://mutax.readthedocs.io/).