Changelog

This document provides a brief summary of changes in each released version of powerpax. More information and release builds are also available on the GitHub releases page.

v0.2.1

• Fix bug in checkpoint_chunked_scan() for multidimensional xs inputs.

v0.2.0

• Require Python 3.10 or later.

• Fix bug in sliced_scan() and checkpoint_chunked_scan() for zero length inputs

v0.1.2

Fix deprecation warning from stray use of jax.tree_map.

v0.1.1

Fix bug in checkpoint_chunked_scan(). The issue caused incorrect results in cases where xs is None and chunk_size does not evenly divide length.

v0.1.0

Initial release, includes:

• chunked_vmap()

• sliced_scan()

• checkpoint_chunked_scan()

• Static()