Samuel

tagging embeddings with text

1

Would it be possible to support adding text labels to each element of the embedding visualisation, similarly to how image sprites can currently be attached via the `label_img` argument to...

Minor formatting fix for `jax.nn.one_hot` docstring

1

Adds minor code formatting to the `jax.nn.one_hot` so that it is self-consistent.

Minor fix for quote formatting in the module docstring contained in `flax.core.variables.py` (at present, there is a duplicate set of double quotes).

Minor typo fix in `flax_fundamentals/flax_basics.ipynb`

1

Minor grammar typo fix in the introductory ipython notebook called `flax_basics.ipynb`

Fix minor typo in docstring for `flax.linen.Module`. The typo fix is `submodels` -> `submodules`.

Minor fixes for code formatting and grammar in the `flax.linen.module.variable` docstring.

Fix small grammatical typos and formatting issues that appear in the `flax.linen.Module.init_with_output` docstring. ## Checklist - [x] This PR fixes a minor issue (e.g.: typo or small bug) or improves...