aspannaus

Coupled SDE System Implementation

7

Hi all, thanks for the great library. I'm having an issue implementing a coupled system of SDEs. I'm getting an ```ValueError: `terms` must be a PyTree of `AbstractTerms` (such as...

Randomness suggestion

5

Thanks for putting the library together. Just a suggestion for giving users control over the randomness. What do you think about the `DataLoader` class taking a `jax.random` key as a...