v0.1.2

Fix deprecation warning from stray use of jax.tree_map.

SHA256 Checksums

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

SHA512 Checksums

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

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.

SHA256 Checksums

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

SHA512 Checksums

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

v0.1.0

Initial release, includes:

• chunked_vmap
• sliced_scan
• checkpoint_chunked_scan
• Static

SHA256 Checksums

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

SHA512 Checksums

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