jax-mlx

C++ ★ 0 updated 10d ago

JAX on Apple Silicon GPUs via an open-source PJRT plugin backed by MLX

No plain-English explanation yet — one is being written right now. Check back in a minute.