Archis Joglekar

Hi, thanks for this great project. I have a use case where I need to do a `value_and_grad` inside a transformed function, and run a few steps with an optax...

Saving metrics during solve

8

Hi, thanks again for the library. we have a numerical integration we perform using `lax.scan`. We'd like to port it over to `diffrax`. I wanted to get your thoughts on...

added bbox_inches flag. Defaults to None

1

Related to https://github.com/jbusecke/xmovie/issues/94 This may help users with eliminating whitespace and other alignment related issues.

Feature request for `bbox_inches = "tight"`

1

First, thanks for putting this together. This is great and saves me an incredible amount of time and effort. Is there any interest in putting in a flag for `bbox_inches...

Adds a lot of Lagrangian Rad Hydro code for a 1D radial geometry