jax
94fcd7f2 - Use taggedtuple instead of namedtuple when defining ShardingSpecs

Commit

5 years ago

Use taggedtuple instead of namedtuple when defining ShardingSpecs Because namedtuples don't take the class into account when comparing for equality!

References

Author

apaszke

Committer

apaszke

Parents