diffrax

diffrax copied to clipboard

Obscure error for passing EulerHeun instead of EulerHeun()

3

Hi, I've wasted some time because I made a small mistake, and the error was quite obscure. I passed `EulerHeun` instead of `EulerHeun()` as the solver argument. Minimal example: ```...

rdaems

Infinite loop in PID stepsize controller due to float precision error

3

There is a special case that leads to the same step in PID being recalculated infinitely. This arises in PIDController.adapt_step_size() when t0 is high, dt is clipped to dtmin, and...

philipwijesinghe

Code breaks on an oscillator-like problem

2

Hello, I have a system of two coupled ODEs. I tried several solvers from Diffrax, both non-stiff and stiff, with and without time step control. Every solver falls on this...

evgenykurbatov

Support Events for BacksolveAdjoint

2

Hi! :) I just wanted to ask if its possible to also support Events for the BacksolveAdjoint. I know that its advised against using BacksolveAdjoint by this library in general...

LuggiStruggi

How to enforce non-negativity constraints?

2

I'm solving an ODE system for viral dynamics using `Kvaerno4` with a PID controller. The state variables need to stay non-negative.... I searched the docs and issues but didn't find...

Mycroft-47

Out of Memory: How to reduce the memory cost for calculating the derivatives wrt the input

2

I want to calculate the derivatives of the output of a neural ODE w.r.t. the input. However, I met with memory issues. I think the memory-saving method `RecursiveCheckpointAdjoint` doesn't support...

zhlfmyzhh

Implementation of multirate methods in diffrax

2

Has there been discussion of implementing [multirate methods](https://arxiv.org/abs/2505.20062) in `diffrax`? See Sec. 2.4.3. of the linked article, Fig. 4. By this, I mean methods tailored to the case where `ODETerm`...

BenjaminDAnjou

Diferentiating w.r.t. a LinearInterpolation

3

Hi there I'm working with dynamical systems and have run in a small issue regarding differentiation with respect to control inputs. From my studies of the examples provided I assumed...

JeanCMthomasio

Our of memory : How to optimize the memory use in Diffrax framework?

4

Hello, I wrote a simulator to simulate a ODE system ( with at least 256x256 ODEs in this system), max_steps=1000, solver is Tsit5, adjoints=diffrax.RecursiveCheckpointAdjoint(), however I found the error: Execution...

timnotavailable

Should Real be deleted in custom types?

1

https://github.com/patrick-kidger/diffrax/blob/main/diffrax/_custom_types.py#L136 has all the jaxtyping imports but Real, I assume this wasn't intentional (or if so, why is that the case?)

lockwo