Cristian Garcia

Thanks for the detailed review @andsteing! Will look into your comments.

@andsteing can you please take another look at this? 🙏 Thanks!

Thanks @andsteing for the detailed review!

@BoyuanJackChen I don't know if I am being too naive, but what if you just batch the data outside of Jax? I mean, just create a generator that produces numpy...

@BoyuanJackChen tf.data uses a lot of tricks to keep the GPU busy, in particular the idea is to do the preprocessing and batching on the CPU in parallel to the...

Hey @conceptofmind! #2367 will move the Community Examples to `docs/examples.rst` so they are visible in the new [Example](https://flax.readthedocs.io/en/latest/examples.html) section on the documentation, this should be ported there (depending on which...

### Update Currently @avital's example does not work: ```python resnet = ResNet() resnet.setup() resnet.backbone # does this work? ``` You now get the following error: ``` AssertionError: Trying to register...

@levskaya should we close this issue?

Hey @vasudevgupta7! We might be able to reference it in [Community Examples](https://github.com/google/flax/blob/main/examples/README.md#community-examples) section. Feel free to send a PR or share it.

Hey @RocketLL, thanks for minimal repro. @jheek @marcvanzee the `Cell` Module (`ScanLSTMCell_0`) is neither reporting its outputs nor the output of its submodules as shown here: ``` { intermediates: {...