Pallas emulation
#20193
Replies: 2 comments 1 reply
-
Apologies for the noise, but I am also curious about emulating Pallas on CPU. Did you manage to figure it out? Thanks! |
Beta Was this translation helpful? Give feedback.
0 replies
-
Try passing |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
The Pallas documentation describes an "Emulation Mode": https://jax.readthedocs.io/en/latest/pallas/design.html#emulation-mode
It doesn't come with an example or any other reference, though. Is this a feature I could use right now? Or is it something that is supported by the design, but not yet implemented?
Beta Was this translation helpful? Give feedback.
All reactions