Fabian Pedregosa

one more thing: could you please add the tag `.. versionchanged:: 0.2.4` to the docstrings of the functions you've changed explaining the change? See for example here for an example:...

also, the new kwargs should be described in the function docstring (on top of adding the `versionchanged` tag)

thanks, we're almost there. Please add type annotations for these two new kwargs. These are likely ``` where: chex.Array | None = None, axis: int | tuple[int, ...] | None...

Solved in https://github.com/google-deepmind/optax/pull/1032

excellent @Abhinavcode13 ! which task would you like to pick up? I would suggest to pick one of the remaining 3 and start with that. Smaller PR are better for...

seems that is pytype now that isn't compatible with python 3.12 ...

@carlosgmartin no updates, this PR is currently orphaned. Do you want to take over?

1. A first step would be to update with main, currently there are some conflicts with the current head. 2. I believe the issues highlighted by @mkunesch only apply to...

excellent, closing this PR then

ouch! yeah, I'm seeing from https://github.com/google/grain/blob/main/pyproject.toml that they require Python 3.10 ....