# jax-metal
The jax-metal package is a Metal GPU plugin to provide Metal acceleration on Mac platforms for JAX applications.
## Installation
The following table tracks `jax-metal` versions and compatible versions of `jax`, `jaxlib` and `MacOS`.
| jax-metal | MacOS | jaxlib | jax |
|-----------|-----------------------------------|---------|---------|
| 0.0.5 | Sonoma 14.2+ | v0.4.20 | v0.4.20 |
| 0.0.4 | Sonoma 14.0+ | v0.4.11 | v0.4.11 |
| 0.0.3 | Ventura 13.4.1+, Sonoma 14.0 Beta | v0.4.10 | v0.4.11 |
We recommend to install the binary package with venv or conda.
```
python3 -m venv ~/jax-metal
source ~/jax-metal/bin/activate
python -m pip install -U pip
python -m pip install numpy wheel
python -m pip install jax-metal
```
Please refer to https://developer.apple.com/metal/jax/ for the full setup and verification.
Raw data
{
"_id": null,
"home_page": "https://developer.apple.com/metal/jax/",
"name": "jax-metal",
"maintainer": "",
"docs_url": null,
"requires_python": "",
"maintainer_email": "",
"keywords": "jax metal plugin",
"author": "",
"author_email": "",
"download_url": "",
"platform": null,
"description": "# jax-metal\n\nThe jax-metal package is a Metal GPU plugin to provide Metal acceleration on Mac platforms for JAX applications.\n\n## Installation\n\nThe following table tracks `jax-metal` versions and compatible versions of `jax`, `jaxlib` and `MacOS`.\n\n| jax-metal | MacOS | jaxlib | jax |\n|-----------|-----------------------------------|---------|---------|\n| 0.0.5 | Sonoma 14.2+ | v0.4.20 | v0.4.20 |\n| 0.0.4 | Sonoma 14.0+ | v0.4.11 | v0.4.11 |\n| 0.0.3 | Ventura 13.4.1+, Sonoma 14.0 Beta | v0.4.10 | v0.4.11 |\n\nWe recommend to install the binary package with venv or conda.\n```\npython3 -m venv ~/jax-metal\nsource ~/jax-metal/bin/activate\npython -m pip install -U pip\npython -m pip install numpy wheel\npython -m pip install jax-metal\n```\n\nPlease refer to https://developer.apple.com/metal/jax/ for the full setup and verification. \n\n",
"bugtrack_url": null,
"license": "Copyright \u00a9 2023 Apple Inc. All rights reserved.",
"summary": "JAX acceleration for Mac GPUs.",
"version": "0.0.5",
"project_urls": {
"Homepage": "https://developer.apple.com/metal/jax/"
},
"split_keywords": [
"jax",
"metal",
"plugin"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "3905213194270f3836fe81f6220f5f23a9dc3132cb3a832547af529059fb4514",
"md5": "080da2901cab0fd70192637d66c4c694",
"sha256": "08059fe8d12897150bf49798b7cded9e1afe9953a9fd710a32e2713f301fee8c"
},
"downloads": -1,
"filename": "jax_metal-0.0.5-py3-none-macosx_10_14_x86_64.whl",
"has_sig": false,
"md5_digest": "080da2901cab0fd70192637d66c4c694",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 54637112,
"upload_time": "2023-12-21T20:31:19",
"upload_time_iso_8601": "2023-12-21T20:31:19.689262Z",
"url": "https://files.pythonhosted.org/packages/39/05/213194270f3836fe81f6220f5f23a9dc3132cb3a832547af529059fb4514/jax_metal-0.0.5-py3-none-macosx_10_14_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "8e15e0c6c864b5a8210d7c0bec34a567b3b8d6071422412333b7fff24c324bea",
"md5": "7a4249ea5c248b9c224f94d3538bcd5c",
"sha256": "565e211628059ca44021349ec36e514bcbfcbf33af610b2912829569bf78b596"
},
"downloads": -1,
"filename": "jax_metal-0.0.5-py3-none-macosx_11_0_arm64.whl",
"has_sig": false,
"md5_digest": "7a4249ea5c248b9c224f94d3538bcd5c",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 40334195,
"upload_time": "2023-12-21T20:31:43",
"upload_time_iso_8601": "2023-12-21T20:31:43.461493Z",
"url": "https://files.pythonhosted.org/packages/8e/15/e0c6c864b5a8210d7c0bec34a567b3b8d6071422412333b7fff24c324bea/jax_metal-0.0.5-py3-none-macosx_11_0_arm64.whl",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2023-12-21 20:31:19",
"github": false,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"lcname": "jax-metal"
}