Powerpax

Fix deprecation warning from stray use of `jax.tree_map`.

<details><summary>SHA256 Checksums</summary>


64cc6843a68286a5ed3557e2b5b55cd2266b2064a8067d97fae1adb059eae9b8 powerpax-0.1.2.tar.gz
22ec238ebe45ea840b3066baf876d57fc2de2cb366b88c0a6f262dde73cf3186 powerpax-0.1.2-py3-none-any.whl


</details>

<details><summary>SHA512 Checksums</summary>


18ca98e1aa6b2de114a2b4d1ace2a28ff64a51b40344a0c3329a74bef28e50cdaa079f3d8f7edaa99a19cdd96a13e1c2e39735d062f79912c82adf31884c5cd3 powerpax-0.1.2.tar.gz
8adcd4213150f0d14f3e9ab0d6e162c8578b28c94146c31d67d71038bd9882e64b7a030517519558380c4721eea6f7ae93372fd24fca42c7a38fe4ac8d2863b5 powerpax-0.1.2-py3-none-any.whl


</details>

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`.

<details><summary>SHA256 Checksums</summary>


caf53b157b18947529e2902dadec877c66469daca87063fe2c4aba7a20cc01bf powerpax-0.1.1.tar.gz
4b040f475059d83902bd275cb1129c90511ade23fa47a3b1fd77e66562d43f85 powerpax-0.1.1-py3-none-any.whl


</details>

<details><summary>SHA512 Checksums</summary>


f4f0f8eca9b83cd44cc37abf9db9c96459ab795a445ed40a5fd2c15cbf43e63286fdc2547b2deeaecfbe7f3a7b445c57278a16f770aead8ad4c8ca3868aed34c powerpax-0.1.1.tar.gz
79108633e7cf2d3b4429a22b927f77bd14b32139c1dd53d6874f249de594bfe29ada24108512d705a01ba54e3683c1375c422322433337b3ef5a3e6805365e1f powerpax-0.1.1-py3-none-any.whl


</details>

Initial release, includes:
- `chunked_vmap`
- `sliced_scan`
- `checkpoint_chunked_scan`
- `Static`


<details><summary>SHA256 Checksums</summary>


f4f2a40a3531b9d2aae5c6fcdb5016fcf4f8bf6c384d78a29247ea5b831df328 powerpax-0.1.0.tar.gz
06437c49c51f05e28d86bf1214c39a5aff0963226d08d46ba8dfbc90c7d237e1 powerpax-0.1.0-py3-none-any.whl


</details>

<details><summary>SHA512 Checksums</summary>


628e7c3ee8c4f7f765192b9d53919026cd8d009e4d766c2ae1c26deda7d7921efe45a5bb39ce3f512662b253a4582f664b008f2ff93987b764e4b3f47f3863ab powerpax-0.1.0.tar.gz
3ac6497d0b94e434d33a1ed6b499d5ab6fe76731f12840303ae3ca29ffb1012cc7b0256af79ded95966dddf5061b8580df483c028bdca23af9017efd943a41e4 powerpax-0.1.0-py3-none-any.whl


</details>