Skip to content

jx.integrate slow and internal dynamics #563

Answered by VENOM314
VENOM314 asked this question in Q&A
Discussion options

You must be logged in to vote

There are a few ODE solving algorithms that different simulators use. For NEURON, they used "thomas". Different algorithms have different performance under different conditions: thomas is slightly faster on CPU, stone is faster on GPU and works best when there are long branches with many compartments (see Jaxley's doc for detailed explaination). For the current cell I am working with, "sparse" provides a short runtime. The lesson learned is to change the solvers/algorithms.

The syntax is jx.integrate(…, voltage_solver=“jax.sparse”)

Replies: 2 comments

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Answer selected by VENOM314
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
1 participant