Using CPython's Frame API?
#11120
Replies: 2 comments
-
|
Hmmm, IIUC, this API only serves as an alternative to tracer. |
Beta Was this translation helpful? Give feedback.
0 replies
-
|
TorchDynamo now support dynamic shape in PyTorch 2.1. But jax jit needs static shape. Any insights on that? |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
TorchDynamo is a new compilation pathway for Pytorch which leverages CPython's Frame API. This API is purpose built for alternative compilers and other things like that.
I'm still getting to know Jax, but looks like there are different ways of dealing with control flow, each with their own tradeoffs. Could a Torch Dynamo like approach be helpful in improving the situation so for example you don't have to use
jax.lax.while_loop, or would you still need some way to express those invariants, unless the compiler can pick up on it?Beta Was this translation helpful? Give feedback.
All reactions