flax
Add out_sharding docstrings to linear layer call methods
#5154
Open

Add out_sharding docstrings to linear layer call methods #5154

jackopenn wants to merge 1 commit into google:main from jackopenn:linear-out-sharding-docs
jackopenn
jackopenn Add out_sharding docstrings to linear layer call methods
2a49924b

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone