Nnx

What's Changed
* Lifted Transforms refactor by cgarciae in https://github.com/cgarciae/nnx/pull/24


**Full Changelog**: https://github.com/cgarciae/nnx/compare/0.0.7...0.0.8

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


**Full Changelog**: https://github.com/cgarciae/nnx/compare/0.0.6...0.0.7

What's Changed
* Generalize PureModule + Scan by cgarciae in https://github.com/cgarciae/nnx/pull/6


**Full Changelog**: https://github.com/cgarciae/nnx/compare/0.0.5...0.0.6

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


**Full Changelog**: https://github.com/cgarciae/nnx/compare/0.0.4...0.0.5

**Major update**
- Simplified APIs
- Modules are now regular mutable python classes
- Added support for Module graphs (RIP pytrees)

NNX is now more pythonic, simpler, and still as powerful as Flax.