Code

Save fix for t0==t1#494

Code

Merged

patrick-kidger merged 22 commits intopatrick-kidger:devpatrick-kidger/diffrax:devfrom dkweiss31:save_fixdkweiss31/diffrax:save_fixCopy head branch name to clipboardDec 6, 2024

+132

Commits

Commits on Sep 1, 2024

• Langevin PR (patrick-kidger#453)

  Show description for fa7417f
  andyElking

  authored
  fa7417f

  Copy full SHA for fa7417f

Commits on Sep 14, 2024

• tidy-ups

  patrick-kidger

  committed
  0f809d0

  Copy full SHA for 0f809d0

Commits on Oct 21, 2024

• Split SDE tests in half, to try and avoid GitHub runner issues?

  patrick-kidger

  committed
  d80ff7a

  Copy full SHA for d80ff7a

• Added effects_barrier to fix test issue with JAX 0.4.33+

  patrick-kidger

  committed
  cbf04ff

  Copy full SHA for cbf04ff

• small fix of docs in all three and a return type in quicsort

  

  andyElking

  authored and

  patrick-kidger

  committed
  2e7836d

  Copy full SHA for 2e7836d

Commits on Nov 2, 2024

• bump doc building pipeline

  patrick-kidger

  committed
  8b34a1c

  Copy full SHA for 8b34a1c

Commits on Nov 18, 2024

• Compatibility with JAX 0.4.36, which removes ConcreteArray

  patrick-kidger

  committed
  161f2a6

  Copy full SHA for 161f2a6

• using a fori_loop to save states in edge case t0==t1

  dkweiss31

  committed
  c88305d

  Copy full SHA for c88305d

• added case for saving t0 data, which was also not getting updated.

  Show description for c96ee56
  dkweiss31

  committed
  c96ee56

  Copy full SHA for c96ee56

• using while_loop, ran into issues with reverse-mode diff using the fori_loop

  dkweiss31

  committed
  439887c

  Copy full SHA for 439887c

• bug fix for cases when t0=True

  dkweiss31

  committed
  dc0dba4

  Copy full SHA for dc0dba4

Commits on Nov 27, 2024

• simplified logic for saving, no loop necessary

  dkweiss31

  committed
  c994a60

  Copy full SHA for c994a60

• added vmap test

  dkweiss31

  committed
  78289c9

  Copy full SHA for 78289c9

Commits on Nov 29, 2024

• using a fori_loop to save states in edge case t0==t1

  

  dkweiss31

  authored and

  patrick-kidger

  committed
  666948c

  Copy full SHA for 666948c

• added case for saving t0 data, which was also not getting updated.

  Show description for 1bd4e08
  

  dkweiss31

  authored and

  patrick-kidger

  committed
  1bd4e08

  Copy full SHA for 1bd4e08

• using while_loop, ran into issues with reverse-mode diff using the fori_loop

  

  dkweiss31

  authored and

  patrick-kidger

  committed
  f23456e

  Copy full SHA for f23456e

• bug fix for cases when t0=True

  

  dkweiss31

  authored and

  patrick-kidger

  committed
  0e00411

  Copy full SHA for 0e00411

• simplified logic for saving, no loop necessary

  

  dkweiss31

  authored and

  patrick-kidger

  committed
  065fe11

  Copy full SHA for 065fe11

• added vmap test

  

  dkweiss31

  authored and

  patrick-kidger

  committed
  47880be

  Copy full SHA for 47880be

Commits on Dec 5, 2024

• Merge remote-tracking branch 'origin/save_fix' into save_fix

  dkweiss31

  committed
  bb292b7

  Copy full SHA for bb292b7

• fix t1 out of bounds issue

  dkweiss31

  committed
  c8ca285

  Copy full SHA for c8ca285

Commits on Dec 6, 2024

• fix for steps: don't want to update those values if t0==t1 since we didn't take any steps.

  Show description for 059f72b
  dkweiss31

  committed
  059f72b

  Copy full SHA for 059f72b