flax
Add out_sharding arguments to linear layers where supported
#5156

Merged

Commits

Add out_sharding arguments to linear layers where supported

samanklesaria committed 96 days ago
fix: update expected shape in conv sharding test

samanklesaria committed 96 days ago

Loading