Closed mfkasim1 closed 3 months ago
solve_idae
seq1d
solve_ivp
jax.lax.slice
jax.lax.slice_in_dim
jax.lax.associative_scan
__getitem__
solve_idae
for solving implicit differential algebraic equations (IDAE)seq1d
,solve_ivp
,solve_idae
jax.lax.slice
(orjax.lax.slice_in_dim
used injax.lax.associative_scan
) by using direct indexing (i.e.,__getitem__
) #16