brainpy.running module#

This module provides APIs for parallel brain simulations.

Parallel Processing 1#

cpu_ordered_parallel(func, arguments[, ...])

Performs a parallel ordered map with a progress bar.

cpu_unordered_parallel(func, arguments[, ...])

Performs a parallel unordered map with a progress bar.

Parallel Processing 2#

process_pool(func, all_params, num_process)

Run multiple models in multi-processes.

process_pool_lock(func, all_params, num_process)

Run multiple models in multi-processes with lock.

Parallel Processing 3#

jax_vectorize_map(func, arguments, num_parallel)

Perform a vectorized map of a function by using jax.vmap.

jax_parallelize_map(func, arguments, ...[, ...])

Perform a parallelized map of a function by using jax.pmap.

Runners#

Runner(target[, monitors, fun_monitors, ...])

Base Runner.