jax-mps
JAX backend for Apple Metal Performance Shaders (MPS)
Installation
In a virtualenv (see these instructions if you need to create one):
pip3 install jax-mps
Dependencies
- None
Releases
| Version | Released | Bullseye Python 3.9 |
Bookworm Python 3.11 |
Trixie Python 3.13 |
Files |
|---|---|---|---|---|---|
| 0.9.13.dev520 pre-release | 2026-03-31 | ||||
|
|
|||||
| 0.9.12 | 2026-03-31 | ||||
|
|
|||||
| 0.9.12.dev518 pre-release | 2026-03-31 | ||||
|
|
|||||
| 0.9.12.dev512 pre-release | 2026-03-29 | ||||
|
|
|||||
| 0.9.12.dev509 pre-release | 2026-03-27 | ||||
|
|
|||||
| 0.9.12.dev508 pre-release | 2026-03-27 | ||||
|
|
|||||
| 0.9.12.dev505 pre-release | 2026-03-27 | ||||
|
|
|||||
| 0.9.12.dev504 pre-release | 2026-03-27 | ||||
|
|
|||||
| 0.9.12.dev496 pre-release | 2026-03-25 | ||||
|
|
|||||
| 0.9.11 | 2026-03-25 | ||||
|
|
|||||
| 0.9.11.dev494 pre-release | 2026-03-25 | ||||
|
|
|||||
| 0.9.11.dev491 pre-release | 2026-03-25 | ||||
|
|
|||||
| 0.9.11.dev487 pre-release | 2026-03-25 | ||||
|
|
|||||
| 0.9.11.dev481 pre-release | 2026-03-25 | ||||
|
|
|||||
| 0.9.11.dev480 pre-release | 2026-03-25 | ||||
|
|
|||||
| 0.9.11.dev473 pre-release | 2026-03-24 | ||||
|
|
|||||
| 0.9.11.dev467 pre-release | 2026-03-23 | ||||
|
|
|||||
| 0.9.11.dev463 pre-release | 2026-03-23 | ||||
|
|
|||||
| 0.9.11.dev454 pre-release | 2026-03-23 | ||||
|
|
|||||
| 0.9.11.dev451 pre-release | 2026-03-22 | ||||
|
|
|||||
| 0.9.11.dev446 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10 | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev444 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev438 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev430 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev426 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev425 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev422 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev421 pre-release | 2026-03-20 | ||||
|
|
|||||
| 0.9.10.dev420 pre-release | 2026-03-19 | ||||
|
|
|||||
| 0.9.10.dev417 pre-release | 2026-03-19 | ||||
|
|
|||||
| 0.9.10.dev413 pre-release | 2026-03-19 | ||||
|
|
|||||
| 0.9.10.dev405 pre-release | 2026-03-18 | ||||
|
|
|||||
| 0.9.10.dev400 pre-release | 2026-03-17 | ||||
|
|
|||||
| 0.9.10.dev394 pre-release | 2026-03-16 | ||||
|
|
|||||
| 0.9.10.dev393 pre-release | 2026-03-16 | ||||
|
|
|||||
| 0.9.10.dev392 pre-release | 2026-03-16 | ||||
|
|
|||||
| 0.9.10.dev388 pre-release | 2026-03-15 | ||||
|
|
|||||
| 0.9.10.dev387 pre-release | 2026-03-15 | ||||
|
|
|||||
| 0.9.10.dev384 pre-release | 2026-03-15 | ||||
|
|
|||||
| 0.9.10.dev383 pre-release | 2026-03-15 | ||||
|
|
|||||
| 0.9.10.dev380 pre-release | 2026-03-14 | ||||
|
|
|||||
| 0.9.10.dev376 pre-release | 2026-03-14 | ||||
|
|
|||||
| 0.9.10.dev371 pre-release | 2026-03-14 | ||||
|
|
|||||
| 0.9.10.dev363 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.10.dev360 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.10.dev357 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.9 | 2026-03-13 | ||||
|
|
|||||
| 0.9.9.dev355 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.9.dev353 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.8 yanked | 2026-03-13 | ||||
|
|
|||||
| 0.9.8.dev351 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.8.dev348 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.8.dev339 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.8.dev338 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.8.dev337 pre-release | 2026-03-13 | ||||
|
|
|||||
| 0.9.8.dev331 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.8.dev330 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.8.dev328 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.8.dev325 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.8.dev320 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.8.dev303 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.7 | 2026-03-12 | ||||
|
|
|||||
| 0.9.7.dev301 pre-release | 2026-03-12 | ||||
|
|
|||||
| 0.9.7.dev298 pre-release | 2026-03-11 | ||||
|
|
|||||
| 0.9.7.dev288 pre-release | 2026-03-04 | ||||
|
|
|||||
| 0.9.7.dev286 pre-release | 2026-03-04 | ||||
|
|
|||||
| 0.9.7.dev283 pre-release | 2026-03-04 | ||||
|
|
|||||
| 0.9.7.dev272 pre-release | 2026-02-24 | ||||
|
|
|||||
| 0.9.7.dev257 pre-release | 2026-02-23 | ||||
|
|
|||||
| 0.9.7.dev233 pre-release | 2026-02-17 | ||||
|
|
|||||
| 0.9.7.dev229 pre-release | 2026-02-17 | ||||
|
|
|||||
| 0.9.7.dev217 pre-release | 2026-02-15 | ||||
|
|
|||||
| 0.9.6 | 2026-02-15 | ||||
|
|
|||||
| 0.9.6.dev214 pre-release | 2026-02-15 | ||||
|
|
|||||
| 0.9.6.dev206 pre-release | 2026-02-14 | ||||
|
|
|||||
| 0.9.6.dev191 pre-release | 2026-02-12 | ||||
|
|
|||||
| 0.9.6.dev181 pre-release | 2026-02-11 | ||||
|
|
|||||
| 0.9.5 | 2026-02-11 | ||||
|
|
|||||
| 0.9.5.dev179 pre-release | 2026-02-11 | ||||
|
|
|||||
| 0.9.5.dev177 pre-release | 2026-02-11 | ||||
|
|
|||||
| 0.9.5.dev174 pre-release | 2026-02-11 | ||||
|
|
|||||
| 0.9.5.dev172 pre-release | 2026-02-11 | ||||
|
|
|||||
| 0.9.4 | 2026-01-30 | ||||
|
|
|||||
| 0.9.3 | 2026-01-29 | ||||
|
|
|||||
| 0.9.2 | 2026-01-28 | ||||
|
|
|||||
| 0.9.1 | 2026-01-27 | ||||
|
|
|||||
| 0.9.0 | 2026-01-26 | ||||
|
|
|||||
Issues with this package?
- Search issues for this package
- Package or version missing? Open a new issue
- Something else? Open a new issue