Code

Stateful now supports vmap#466

Code

Merged

patrick-kidger merged 3 commits intodevpatrick-kidger/equinox:devfrom better-statefulpatrick-kidger/equinox:better-statefulCopy head branch name to clipboardSep 12, 2023

+498-61

Commits

Commits on Sep 12, 2023

• Updated stateful operations to be able to support vmap'ing.

  Show description for 2d4c553
  patrick-kidger

  committed
  2d4c553

  Copy full SHA for 2d4c553

• Stateful operations now support creating states multiple times, such that they are compatible with the original model.

  patrick-kidger

  committed
  61e47b8

  Copy full SHA for 61e47b8

• Added deprecation message

  patrick-kidger

  committed
  3c8d4e8

  Copy full SHA for 3c8d4e8