polysolve 0.7.1

A Python library for representing, manipulating, and solving polynomial equations. Features a high-performance, Numba-accelerated genetic algorithm for CPU, with an optional CUDA/GPU backend for massive-scale parallel solving.

Key Features

• Numerically Stable Solver: Makes complex calculations practical. Leverage your GPU to power the robust genetic algorithm, solving high-degree polynomials accurately in a reasonable timeframe.
• Complex Number Support: Fully supports complex coefficients and finding roots in the complex plane (e.g., $x^2 + 1 = 0 \to \pm i$).
• Numba Accelerated CPU Solver: The default genetic algorithm is JIT-compiled with Numba for high-speed CPU performance, right out of the box.
• CUDA Accelerated: Leverage NVIDIA GPUs for a massive performance boost when finding roots in large solution spaces.
• Create and Manipulate Polynomials: Easily define polynomials of any degree using integer or float coefficients, and perform arithmetic operations like addition, subtraction, multiplication, and scaling.
• Analytical Solvers: Includes standard, exact solvers for simple cases (e.g., quadratic_solve).
• Simple API: Designed to be intuitive and easy to integrate into any project.

Installation

Install the base package from PyPI:

pip install polysolve

CUDA Acceleration

To enable GPU acceleration, install the extra that matches your installed NVIDIA CUDA Toolkit version. This provides a significant speedup for the genetic algorithm.

For CUDA 12.x users:

pip install polysolve[cuda12]

Quick Start

Here is a simple example of how to define a quadratic function, find its properties, and solve for its roots.

from polysolve import Function, GA_Options

# 1. Define the function f(x) = 2x^2 - 3x - 5
#    Coefficients can be integers or floats.
f1 = Function(largest_exponent=2)
f1.set_coeffs([2, -3, -5])

print(f"Function f1: {f1}")
# > Function f1: 2x^2 - 3x - 5

# 2. Solve for y at a given x
y_val = f1.solve_y(5)
print(f"Value of f1 at x=5 is: {y_val}")
# > Value of f1 at x=5 is: 30.0

# 3. Get the derivative: 4x - 3
df1 = f1.derivative()
print(f"Derivative of f1: {df1}")
# > Derivative of f1: 4x - 3

# 4. Get the 2nd derivative: 4
ddf1 = f1.nth_derivative(2)
print(f"2nd Derivative of f1: {ddf1}")
# > Derivative of f1: 4

# 5. Find roots analytically using the quadratic formula
#    This is exact and fast for degree-2 polynomials.
roots_analytic = f1.quadratic_solve()
print(f"Analytic roots: {sorted(roots_analytic)}")
# > Analytic roots: [-1.0, 2.5]

# 6. Find REAL roots with the genetic algorithm (Numba CPU)
#    This is the default, JIT-compiled CPU solver.
ga_opts = GA_Options(num_of_generations=20)
roots_ga = f1.get_real_roots(ga_opts, use_cuda=False)
print(f"Approximate real roots: {roots_ga[:2]}")
# > Approximate real roots: [-1.000..., 2.500...]

# 7. Find ALL roots (Real + Complex)
#    Use get_roots() to search the complex plane.
f_complex = Function(2, [1, 0, 1]) # x^2 + 1
roots_all = f_complex.get_roots(ga_opts)
print(f"Approximate complex roots: {roots_all}")
# > Approximate complex roots: [-1.00...j, 1.00...j]

# If you installed a CUDA extra, you can run it on the GPU:
# roots_ga_gpu = f1.get_real_roots(ga_opts, use_cuda=True)
# print(f"Approximate roots from GA (GPU): {roots_ga_gpu[:2]}")

Tuning the Genetic Algorithm

The GA_Options class gives you fine-grained control over the genetic algorithm's performance, letting you trade speed for accuracy.

The default options are balanced, but for very complex polynomials, you may want a more exhaustive search.

from polysolve import GA_Options

# Create a config for a deep search, optimized for finding
# *all* real roots (even if they are far apart).
ga_robust_search = GA_Options(
    num_of_generations=50,  # Run for more generations
    data_size=500000,       # Use a larger population

    # --- Key Tuning Parameters for Multi-Root Finding ---

    # Widen the parent pool to 75% to keep more "niches"
    # (solution-clouds around different roots) alive.
    selection_percentile=0.75, 

    # Increase the crossover blend factor to 0.75.
    # This allows new solutions to be created further
    # away from their parents, increasing exploration.
    blend_alpha=0.75,

    # Enable complex root finding (default is True)
    find_complex=True
)

# Pass the custom options to the solver
roots = f1.get_real_roots(ga_accurate)

For a full breakdown of all parameters, including crossover_ratio, mutation_strength, and more, please see the full GA_Options API Documentation.

Development & Testing Environment

This project is automatically tested against a specific set of dependencies to ensure stability. Our Continuous Integration (CI) pipeline runs on an environment using CUDA 12.5 on Ubuntu 24.04.

While the code may work on other configurations, all contributions must pass the automated tests in our reference environment. For detailed information on how to replicate the testing environment, please see our Contributing Guide.

Contributing

Contributions are welcome! Whether it's a bug report, a feature request, or a pull request, please feel free to get involved.

Please read our CONTRIBUTING.md file for details on our code of conduct and the process for submitting pull requests.

Contributors

Jonathan Rampersad 🚧 💻 📖 🚇

Add your contributions

License

This project is licensed under the MIT License - see the LICENSE file for details.