What's Changed

• Replace deprecated jax.tree_* functions with jax.tree.* by @copybara-service in #135
• jax.numpy.clip: update use of deprecated arguments. by @copybara-service in #138
• Configure tests to explicitly use jax_threefry_partitionable=False. by @copybara-service in #141
• Move to python 3.12 and 3.13, part 2. by @copybara-service in #144
• Update pypi workflow. by @copybara-service in #143

Full Changelog: v0.1.6...v0.1.7