Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
No more tuples #1224
This PR removes tuples from the jaxpr language, and JaxTuples / AbstractTuples / DeviceTuples / ShardedDeviceTuples from the implementation. Instead of tuples, jaxprs and primitives have multiple outputs.
Here are the main reasons for removing tuples:
So this PR gets rid of them! We also took the opportunity to:
I checked for performance regressions on
Two things that will probably come in smaller follow-up PRs: