# jax-scipy-spatial
This package implements `scipy.spatial` API for JAX.
Currently the following items are implemented:
- `scipy.spatial.distance`
- `scipy.spatial.transform.Rotation`
- `scipy.spatial.transform.Slerp`
## Install
```
pip install .
```
## Usage
```
import jax_scipy_spatial.transform as jtr
rotation = jtr.Rotation.from_euler('xyz', jnp.array([0., 0., 180.]), degrees=True)
print(rotation)
```
## Documentation
Please refer to [scipy documentation](https://docs.scipy.org/doc/scipy/reference/spatial.html).
## Contributing
To run unit tests on your local machine:
```
tox
```
Raw data
{
"_id": null,
"home_page": null,
"name": "jax-scipy-spatial",
"maintainer": null,
"docs_url": null,
"requires_python": null,
"maintainer_email": null,
"keywords": "jax, scipy, spatial, rotation",
"author": null,
"author_email": "Chris Flesher <chris.flesher@gmail.com>",
"download_url": "https://files.pythonhosted.org/packages/77/6f/d05a728b797f841362a15c1cc1c572958b2ca3560855a33032d14405a15c/jax-scipy-spatial-0.2.3.tar.gz",
"platform": null,
"description": "# jax-scipy-spatial\n\nThis package implements `scipy.spatial` API for JAX.\n\nCurrently the following items are implemented:\n\n- `scipy.spatial.distance`\n- `scipy.spatial.transform.Rotation`\n- `scipy.spatial.transform.Slerp`\n\n## Install\n```\npip install .\n```\n\n## Usage\n```\nimport jax_scipy_spatial.transform as jtr\n\nrotation = jtr.Rotation.from_euler('xyz', jnp.array([0., 0., 180.]), degrees=True)\nprint(rotation)\n```\n\n## Documentation\nPlease refer to [scipy documentation](https://docs.scipy.org/doc/scipy/reference/spatial.html).\n\n## Contributing\n\nTo run unit tests on your local machine:\n```\ntox\n```\n",
"bugtrack_url": null,
"license": null,
"summary": "Scipy spatial API for JAX",
"version": "0.2.3",
"project_urls": {
"Bug Tracker": "https://github.com/chrisflesher/jax-scipy-spatial/issues",
"Homepage": "https://github.com/chrisflesher/jax-scipy-spatial"
},
"split_keywords": [
"jax",
" scipy",
" spatial",
" rotation"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "23bd781ae308c222f5d4d9eaf885ff66633f7e73579fdff8b1185f52d2809885",
"md5": "c41ac5290085c0070b068e63d90d5d0e",
"sha256": "0a8bf72aff329faed8c2c4de9292157f5679f01fab4ccf44fe52671e83d86449"
},
"downloads": -1,
"filename": "jax_scipy_spatial-0.2.3-py3-none-any.whl",
"has_sig": false,
"md5_digest": "c41ac5290085c0070b068e63d90d5d0e",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 14171,
"upload_time": "2024-04-02T18:30:47",
"upload_time_iso_8601": "2024-04-02T18:30:47.838923Z",
"url": "https://files.pythonhosted.org/packages/23/bd/781ae308c222f5d4d9eaf885ff66633f7e73579fdff8b1185f52d2809885/jax_scipy_spatial-0.2.3-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "776fd05a728b797f841362a15c1cc1c572958b2ca3560855a33032d14405a15c",
"md5": "5859160e69631d5d3fa2bf9539eba417",
"sha256": "22b402ffa2ebd553315e8bf9c85dc1fdbd7282cb66389b4ba9ac6e11a36ae78b"
},
"downloads": -1,
"filename": "jax-scipy-spatial-0.2.3.tar.gz",
"has_sig": false,
"md5_digest": "5859160e69631d5d3fa2bf9539eba417",
"packagetype": "sdist",
"python_version": "source",
"requires_python": null,
"size": 17189,
"upload_time": "2024-04-02T18:30:49",
"upload_time_iso_8601": "2024-04-02T18:30:49.604700Z",
"url": "https://files.pythonhosted.org/packages/77/6f/d05a728b797f841362a15c1cc1c572958b2ca3560855a33032d14405a15c/jax-scipy-spatial-0.2.3.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-04-02 18:30:49",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "chrisflesher",
"github_project": "jax-scipy-spatial",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"tox": true,
"lcname": "jax-scipy-spatial"
}
JSON
