Patrick Power

What's the latest status on this project? Seems useful

@cgarciae Thanks for the response. I will make a benchmark this week and get back to you -- much appreciated

@jatentaki I would be interested in helping to work on this. Looks useful/interesting!

@jatentaki You mentioned that you might make a tutorial on this. I think it would be helpful to have more examples/tutorials involving odes. So if you need help with this,...

@levskaya Is there a simpler way to build an "ode-layer" in flax?

> At the moment the framework is difficult to get started with for newcomers. Without reading the code it's not clear what modules there are and how they're supposed to...

@noanabeshima I am experiencing the same issue. The code runs fine though when I paste it into a Google Colab and connect to a local run time. I am not...

@mblondel I would be happy to tackle this sometime over the next week or so if no one else has started working on it.

@younesbelkada I would certainly be interested if bnb could support torch.compile!

Is the code used to produce `memory_ratio` available? That looks helpful too.