rockpool.utilities.benchmarking.lif_benchmarks

Define benchmark functions for LIF layers

Functions

jax_lif_jit_cpu_benchmark()
jax_lif_jit_gpu_benchmark()
jax_lif_jit_tpu_benchmark()
jax_lif_nojit_benchmark()
lif_benchmark()
lif_exodus_cuda_benchmark()
lif_torch_benchmark()
lif_torch_cuda_benchmark()
lif_torch_cuda_graph_benchmark()