Index

Symbols

00_intro module

01_sweep module

A

A (DCMTheta attribute) (JRTheta attribute) a (JRTheta attribute) a_1 (JRTheta attribute)

a_2 (JRTheta attribute) a_3 (JRTheta attribute) a_4 (JRTheta attribute) alpha (BOLDTheta attribute)

B

B (DCMTheta attribute) (JRTheta attribute) b (JRTheta attribute)

bold_dfun() (in module vbjax.neural_mass) BOLDTheta (class in vbjax.neural_mass) bvep_dfun() (in module vbjax.neural_mass) BVEPTheta (class in vbjax.neural_mass)

C

C (DCMTheta attribute) compute_bold_theta() (in module vbjax.neural_mass) compute_sarvas_gain() (in module vbjax.monitor)

cr (MPRTheta attribute) csr_to_jax_bcoo() (in module vbjax.sparse) cv (MPRTheta attribute)

D

dcm_dfun() (in module vbjax.neural_mass) DCMTheta (class in vbjax.neural_mass) Delta (MPRTheta attribute) dopa_dfun() (in module vbjax.neural_mass) dopa_gfun_add() (in module vbjax.neural_mass)

dopa_gfun_mulr() (in module vbjax.neural_mass) dopa_net_dfun() (in module vbjax.neural_mass) dopa_r_positive() (in module vbjax.neural_mass) DopaState (class in vbjax.neural_mass) DopaTheta (in module vbjax.neural_mass) Dp (DopaState attribute)

E

e0 (BOLDTheta attribute)

epsilon (BOLDTheta attribute) eta (MPRTheta attribute)

G

grid_pairwise_distance() (in module vbjax.shtlc)

H

heun_step() (in module vbjax.loops)

I

I (JRTheta attribute) (MPRTheta attribute)

I1 (BVEPTheta attribute)

J

J (JRTheta attribute) (MPRTheta attribute)

jr_dfun() (in module vbjax.neural_mass) JRState (class in vbjax.neural_mass) JRTheta (class in vbjax.neural_mass)

K

k1 (BOLDTheta attribute) k2 (BOLDTheta attribute) k3 (BOLDTheta attribute) kernel_conv_prep() (in module vbjax.shtlc) kernel_diff() (in module vbjax.shtlc)

kernel_dist_origin() (in module vbjax.shtlc) kernel_estimate_shtns() (in module vbjax.shtlc) kernel_gaussian() (in module vbjax.shtlc) kernel_laplace() (in module vbjax.shtlc) kernel_mexican_hat() (in module vbjax.shtlc) kernel_sh_normalized() (in module vbjax.shtlc)

M

make_bold() (in module vbjax.monitor) make_conn_latent_mvnorm() (in module vbjax.connectome) make_continuation() (in module vbjax.loops) make_cov() (in module vbjax.monitor) make_dde() (in module vbjax.loops) make_dense_layers() (in module vbjax.layers) make_fc() (in module vbjax.monitor) make_fft() (in module vbjax.monitor) make_gain() (in module vbjax.monitor) make_grid() (in module vbjax.shtlc) make_grid_shtns() (in module vbjax.shtlc) make_lm() (in module vbjax.shtlc) make_ode() (in module vbjax.loops) make_offline() (in module vbjax.monitor) make_sdde() (in module vbjax.loops) make_sde() (in module vbjax.loops) make_sg_spmv() (in module vbjax.sparse) make_shtdiff() (in module vbjax.shtlc)

make_shtdiff_np() (in module vbjax.shtlc) make_spmv() (in module vbjax.sparse) make_timeavg() (in module vbjax.monitor) module 00_intro 01_sweep vbjax.connectome vbjax.layers vbjax.loops vbjax.monitor vbjax.neural_mass vbjax.shtlc vbjax.sparse vbjax.util mpr_dfun() (in module vbjax.neural_mass) mpr_r_positive() (in module vbjax.neural_mass) MPRState (class in vbjax.neural_mass) MPRTheta (class in vbjax.neural_mass) mu (JRTheta attribute)

N

nu_0 (BOLDTheta attribute)

nu_max (JRTheta attribute)

R

r (DopaState attribute) (JRTheta attribute) (MPRState attribute) r_0 (BOLDTheta attribute) randn_relaxed() (in module vbjax.shtlc)

recip_alpha (BOLDTheta attribute) recip_e0 (BOLDTheta attribute) recip_tau_f (BOLDTheta attribute) recip_tau_o (BOLDTheta attribute) recip_tau_s (BOLDTheta attribute)

S

Sa (DopaState attribute)

Sg (DopaState attribute)

T

tau (MPRTheta attribute) tau0 (BVEPTheta attribute) tau_f (BOLDTheta attribute) tau_o (BOLDTheta attribute) tau_s (BOLDTheta attribute)

te (BOLDTheta attribute) to_jax() (in module vbjax.util) to_np() (in module vbjax.util) tuple_meshgrid() (in module vbjax.util) tuple_ravel() (in module vbjax.util) tuple_shard() (in module vbjax.util)

U

u (DopaState attribute)

V

V (DopaState attribute) (MPRState attribute) v0 (BOLDTheta attribute) (JRTheta attribute) vbjax.connectome module vbjax.layers module vbjax.loops module

vbjax.monitor module vbjax.neural_mass module vbjax.shtlc module vbjax.sparse module vbjax.util module

X

x0 (BVEPTheta attribute)

Y

y0 (JRState attribute) y1 (JRState attribute) y2 (JRState attribute)

y3 (JRState attribute) y4 (JRState attribute) y5 (JRState attribute)