New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Performance on M1 chips compared to PyTorch implementation? #20

Open

andrewschreiber opened this issue Apr 22, 2023 · 1 comment

andrewschreiber commented Apr 22, 2023

Looks like JAX does not support accelerated M1 / Apple Neural Engine.

Curious if anyone has done a benchmark comparison. Reference:

google/jax#8074

openai/whisper#382

Owner

sanchit-gandhi commented Apr 24, 2023

This would be super cool 🙌 Think we need JAX working on M1 first (as per the first link you've shared)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment