flax
ca19b9bc - Add out_sharding argument to Embed layer call

Commit

223 days ago

Add out_sharding argument to Embed layer call

References

#5205 - Add out_sharding argument to Embed layer call

Author

samanklesaria

samanklesaria

Parents

Loading