Ivy Zheng

> Thanks! I guess the only test failing is having more than 5 commits in the merge. I think you should be able to squash and merge and then it...

On a quick look at the torch documentation and the source code of `jax.random.truncated_normal`, it seems that: * the torch version runs on a fixed normal distribution and *redraws a...

Hmm unfortunately I cannot repro this (Flax 0.8.5). My printout yields this: This can be reproed by opening any empty [colab](https://colab.research.google.com/). ``` Foo Summary ┏━━━━━━━━━┳━━━━━━━━┳━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━┳━━━━━━━┳━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━┓ ┃ path ┃ module ┃...

Hi! I am working on the `NNXToLinen` wrapper that allows you to use NNX within Linen. I likely will send out the actual PR in a few days but for...

Somehow I just noticed this but it's good to have! Might just need to restructure the code after we moved the directory.

Closing this PR because the modified line was removed in a recent doc update.

Just wondering - if there is no params, is there a particular reason for having an empty dict for it? Because no params means the module will not look into...

The code is a bit unreadable in its current format, but from the error it seems solvable by passing `mutable=True` in your `apply()` calls. See more in the documentation of...

Hmm, are you using the latest Flax? The latest Flax should have `flax.config.flax_return_frozendict` as `False` and return normal dict by default. In other case, you can do an explicit check...