What's Changed * Remove utils.py by cgarciae in https://github.com/cgarciae/nnx/pull/14 * Single string State paths by cgarciae in https://github.com/cgarciae/nnx/pull/12 * Implement sow by cgarciae in https://github.com/cgarciae/nnx/pull/16 * New TrainState API by cgarciae in https://github.com/cgarciae/nnx/pull/19 * Use custom UUID instead of id() by cgarciae in https://github.com/cgarciae/nnx/pull/13 * Simplify RngStream and Context by cgarciae in https://github.com/cgarciae/nnx/pull/15 * Changes for MaxText by cgarciae in https://github.com/cgarciae/nnx/pull/21 * SPMD Support by cgarciae in https://github.com/cgarciae/nnx/pull/22 * Type-based collections by cgarciae in https://github.com/cgarciae/nnx/pull/23
What's Changed * Add context constructor by cgarciae in https://github.com/cgarciae/nnx/pull/3 * Add Container base class by cgarciae in https://github.com/cgarciae/nnx/pull/4