Skip to content
This repository has been archived by the owner on Feb 26, 2023. It is now read-only.

0.6.3
Compare
Choose a tag to compare
@cgarciae cgarciae released this 08 Nov 19:26
· 44 commits to master since this release
0.6.3
ca72a04

Changes

  • Adds experimental axis_name argument to next_key, KeySeq, and Linear.
  • Creates the preserve_state function that enables you apply a function transformation like jit or vmap to a stateful method that doesn't propagate the state through and output. preserver_state will return the first argument (usually self) and update it after the transformation.
  • Fixes type issues with Filter in shortcut methods.
Assets 2
Loading