brainpy.running module#

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.

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.

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.