cgiovanetti

Can't return solution of coupled differential equations

5

I'm trying to solve a mid-sized system of coupled differential equations with `diffrax`. I'm using version 0.2.0. Here's a short snippet of dummy code that raises the issue I'm having:...

Difficulty installing diffrax on apple silicon

8

I recently ran a hack session for a code that uses diffrax, but had some trouble getting some participants set up with JAX/diffrax, especially those using M1 or M3 macs...

Old versions much faster?

4

I have a [large project](https://github.com/cgiovanetti/LINX/tree/main) that's intended to run on CPU--during development it was running in well under a second per evaluation, but now I find it runs almost twice...