timnotavailable

Solver selection -- multistep/Adams methods

4

Hello, I have a short question about how to select a proper solver: I have seen some projects using CVODE Adams-Moulton formulas with variable steps. I'm searching for available solver...

Questions on eqx.filter_vmap: How to batch multiple function

7

Hello, I have a question regarding eqx.filter_vmap (perhaps also diffrax): How do I effectively wrap a list of eqx.Module so that I have made a sophiscated ODEs system, the structure...

What is the difference between jax.lax.stop_gradient and eqx.static_field=True ?

7

Hello, I'm going to make a ODE simulator for Bloch Equation, I would like to define a module which has some attributes of array type, class(eqx.Module): array1:Array["x y z"] array2:...

Can we use GaussNewton/LevenbergMarquardt on KL loss minimization

3

Hello, suppose I have a small variational autoencoder VAE (under 200 paras) and I'm going to train it in a very time efficient way, so because the learning parameters are...

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...

ODE solver: some time period has algebraic solution

4

Hello Patrick, I met a problem on realizing such function: Suppose I have such ODE function: From t0 to t1 I have dy/dx=f(t,y), in this period of time I have...