Open caseybasichis opened 8 months ago
No plans for this, but all of our code is straight JAX so if JAX supports M3 we should support it too.
Lots of errors running the tests. No Jax experience here, so it's mostly indecipherable.
Sounds like JAX doesn't support whatever hardware you're running on yet.
Very excited this has been released. Will it work on a M3 Max?
If not, any plans to support Macs?