patrick-kidger / patrick-kidger/diffrax

Understand diffrax.diffeqsolve

Open
#467 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
2.1k
Forks
189
Avg merge
3d 18h
Merged PRs (30d)
1

Description

Hi,

Thanks for the nice package. I have a question regarding the implementation of diffrax.diffeqsolve. Specifically, I want to know a bit more details about the ODE solver, especially on what are the key factors driving diffeqsolve to be faster. In a nutshell, can you list several bulletpoints on the code implementation optimizations that have been done to accelerate ODE solving in diffrax?

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reading the implementation of diffrax.diffeqsolve and trace the main ODE-solving path. Identify the implementation optimizations that affect speed, then document them as several concise bullet points; done means the package has a clear explanation addressing the question.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
documentation
Issue type
Documentation
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.