-
|
As tensorflow 2.4 is optimized for m1 is this the case for jax also blogpost |
Beta Was this translation helpful? Give feedback.
Replies: 4 comments 4 replies
-
|
Not at the moment. But we'd welcome contributions that did so! (It's not entirely clear to me what APIs Apple are exposing and whether we can usefully target them from XLA.) |
Beta Was this translation helpful? Give feedback.
-
|
We should probably ask some Tensorflow folks about this :P |
Beta Was this translation helpful? Give feedback.
-
|
Looping in @rxwei cc @skye @jekbradbury |
Beta Was this translation helpful? Give feedback.
-
|
New PR: |
Beta Was this translation helpful? Give feedback.
-
|
Update: This is merged! |
Beta Was this translation helpful? Give feedback.
-
|
Update: issue #5501 (comment) and PR #7268 from @hawkinsp |
Beta Was this translation helpful? Give feedback.
-
|
Beta Was this translation helpful? Give feedback.
-
|
#5501 (comment) by @dfm 👍 . Workflow: https://github.com/dfm/custom-wheels/blob/217352d8dcc311080dacd91f9919d913cdb9f0d2/.github/workflows/jaxlib.yml#L11-L49 |
Beta Was this translation helpful? Give feedback.
Not at the moment. But we'd welcome contributions that did so!
(It's not entirely clear to me what APIs Apple are exposing and whether we can usefully target them from XLA.)