brainpy.running module

brainpy.running module#

jax_vectorize_map

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

jax_parallelize_map

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

process_pool

Run multiple models in multi-processes.

process_pool_lock

Run multiple models in multi-processes with lock.

cpu_ordered_parallel

Performs a parallel ordered map with a progress bar.

cpu_unordered_parallel

Performs a parallel unordered map with a progress bar.