v0.8.4

[cgarciae]

What's Changed

• fixed codecov by @chiamp in fixed codecov #3895
• Make FlatState a Mapping instead of a dict by @NeilGirdhar in Make FlatState a Mapping instead of a dict #3880
• Share nnx node registry between threads by @NeilGirdhar in Share nnx node registry between threads #3901
• fixed jnp.clip deprecation by @chiamp in fixed jnp.clip deprecation #3905
• Added three tab option to sphinx directive codediff and added testing for first tab by @chiamp in Added three tab option to sphinx directive codediff and added testing for first tab #3847
• Add support for jax.sharding.PartitionSpec.UNCONSTRAINED in logical specification by @copybara-service in Add support for jax.sharding.PartitionSpec.UNCONSTRAINED in logical specification #3902
• [nnx] fix mypy and pytype by @cgarciae in [nnx] fix mypy and pytype #3894
• [nnx] fix iter_nodes by @cgarciae in [nnx] fix iter_nodes #3889
• [nnx] Sequential uses regular list by @cgarciae in [nnx] Sequential uses regular list #3909
• [nnx] add ConvTranspose by @cgarciae in [nnx] add ConvTranspose #3908
• [nnx] add Module pytree_experimental static test by @cgarciae in [nnx] add Module pytree_experimental static test #3864
• Added docstring for Module.scope.path by @chiamp in Added docstring for Module.scope.path #3913
• [linen] test jit caching with state updates by @cgarciae in [linen] test jit caching with state updates #3900
• v0.8.4 by @cgarciae in v0.8.4 #3891
• [linen] enable separate initializers for out layer in MultiHeadDotProductAttention by @cgarciae in [linen] enable separate initializers for out layer in MultiHeadDotProductAttention #3835
• [nnx] cleanup graph by @cgarciae in [nnx] cleanup graph #3915
• [nnx] fix bugs by @cgarciae in [nnx] fix bugs #3925
• Replace deprecated jax.tree_* functions with jax.tree.* by @copybara-service in Replace deprecated jax.tree_* functions with jax.tree.* #3926
• [nnx] Object refactor by @cgarciae in [nnx] Object refactor #3910
• [nnx] add iter_graph by @cgarciae in [nnx] add iter_graph #3919
• [nnx] add compat by @cgarciae in [nnx] add compat #3921
• [nnx] transforms refactor by @cgarciae in [nnx] transforms refactor #3927
• added equivalence test for nnx.ConvTranspose by @chiamp in added equivalence test for nnx.ConvTranspose #3934
• added equivalence test for nnx.Sequential by @chiamp in added equivalence test for nnx.Sequential #3935
• [NNX] Add LoRA and LoRALinear to NNX by @IvyZX in [NNX] Add LoRA and LoRALinear to NNX #3929
• [nnx] fix substate mutability by @cgarciae in [nnx] fix substate mutability #3932
• [nnx] improve update context by @cgarciae in [nnx] improve update context #3933
• [nnx] move out of experimental by @cgarciae in [nnx] move out of experimental #3936

Full Changelog: v0.8.3...v0.8.4

---

This discussion was created from the release v0.8.4.