equinox icon indicating copy to clipboard operation
equinox copied to clipboard

Published 20 hours ago • verified publisher icon patrick-kidger

Enhancements to `equinox.internal.while_loop(..., kind="checkpointed")`

Open patrick-kidger opened this issue 1 year ago • 0 comments

  • [ ] Have it support classical treeverse, for improved results when using equinox.internal.scan(..., kind="checkpointed").
  • [ ] Have it support saving the residuals of each step, rather than just the evolving state.
  • [x] Have it support reading from buffers as well as writing to them.

patrick-kidger avatar Mar 09 '23 05:03 patrick-kidger