v0.1.3

• #22 Improves the type check for jax arrays and enables support for Jax 0.4.1 or newer.

v0.1.2

• Add JAX array support to pybaum. By @timmens and @janosg in #18 and #20

v0.1.1

What's Changed

• Add a section about the API to the docs. by @tobiasraabe in #12
• Add default arguments to tree_just_flatten. by @tobiasraabe in #10
• Implement tree_yield and tree_just_yield. by @tobiasraabe in #11
• Add badges and a short description to the readme. by @tobiasraabe in #13

Full Changelog: v0.1.0...v0.1.1

v0.1.0

Release the initial version. (#1)