flax
1c2cec46 - Use with_sharding_constraint for Format application

Commit

114 days ago

Use with_sharding_constraint for Format application

References

#5246 - Generalize out_sharding to work with NamedSharding and Format

Author

samanklesaria

samanklesaria

Committer

samanklesaria

samanklesaria

Parents

Loading