jax
7c2f8423 - shard_map and other fixes to direct-linearize

Commit

336 days ago

shard_map and other fixes to direct-linearize Co-authored-by: Dougal Maclaurin <dougalm@google.com>

References

Author

mattjj

Committer

mattjj

Parents