flax
23a1c1b1 - Add out_sharding arguments to linear layers where supported

Commit

48 days ago

Add out_sharding arguments to linear layers where supported

References

#5156 - Add out_sharding arguments to linear layers where supported

Author

jackopenn

jackopenn

Committer

samanklesaria

samanklesaria

Parents

Loading