From 32afc85f94d64697562aa38bfce6bff592556654 Mon Sep 17 00:00:00 2001 From: Cristian Garcia Date: Mon, 24 Mar 2025 13:28:19 -0700 Subject: [PATCH] roll back #4646 PiperOrigin-RevId: 740051057 --- flax/nnx/README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flax/nnx/README.md b/flax/nnx/README.md index 14829c0fa..84dfb7410 100644 --- a/flax/nnx/README.md +++ b/flax/nnx/README.md @@ -7,8 +7,8 @@ _**N**eural **N**etworks for JA**X**_ - | [docs](https://flax.readthedocs.io/en/ NNX is a JAX-based neural network library that focuses on providing the best development experience to make building and experimenting with neural networks as easy and intuitive as possible. -* **Pythonic**: Modules are standard Python classes, promoting ease of use and a more familiar - development experience. +* **Pythonic**: Modules are standard Python classes, promoting ease of use and a more + familiar development experience. * **Easy-to-use**: NNX provides a set of transforms that take care of state management, allowing users to focus on building their models and training loops. * **Expressive**: NNX allows fine-grained over the Module state with lifted transforms, enabling