mandelbrot_benchmark package¶ Subpackages¶ mandelbrot_benchmark.backends package Submodules mandelbrot_benchmark.backends.jax module mandelbrot_jax() mandelbrot_benchmark.backends.numba module mandelbrot_numba() mandelbrot_benchmark.backends.taichi module mandebrot_taichi() mandelbrot_benchmark.backends.torch module mandelbrot_torch() mandelbrot_benchmark.backends.warp module mandelbrot_warp() Submodules¶ mandelbrot_benchmark.cli module¶ mandelbrot_benchmark.cli.benchmark(backends: str = 'numba,taichi,warp', max_size_cpu: int = 10, max_size_cuda: int = 13, size_step: float = 0.1) → None[source]¶ Run the Mandelbrot benchmark for different backends and devices. mandelbrot_benchmark.cli.plot() → None[source]¶ Plot the results. mandelbrot_benchmark.plot module¶ mandelbrot_benchmark.plot.plot_mandelbrot(z: Any, /, *, ax: Axes | None = None, extent: tuple[float, float, float, float] | None = None) → None[source]¶ Plot the Mandelbrot set.