# 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/cc/eb/592cae2b7d5af29ab1113233bd0c8adb62e9bf6499884745140b422ab3d5/jax_scipy_spatial-0.2.5.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.5",
"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": "047f3c0723d8f88601f1f065c66731b46b41dc0c6860276fe6aa0e7abf4238f6",
"md5": "7867eeeac167b9aeef398c74918b9533",
"sha256": "b04cd69b57718ba50adfb6d5d1f732b3e54d7bc2dd31451ed6da4f872887c84f"
},
"downloads": -1,
"filename": "jax_scipy_spatial-0.2.5-py3-none-any.whl",
"has_sig": false,
"md5_digest": "7867eeeac167b9aeef398c74918b9533",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 13376,
"upload_time": "2024-11-27T16:26:10",
"upload_time_iso_8601": "2024-11-27T16:26:10.996863Z",
"url": "https://files.pythonhosted.org/packages/04/7f/3c0723d8f88601f1f065c66731b46b41dc0c6860276fe6aa0e7abf4238f6/jax_scipy_spatial-0.2.5-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "cceb592cae2b7d5af29ab1113233bd0c8adb62e9bf6499884745140b422ab3d5",
"md5": "cca0aabe2f78ce464b6b93ab77ac41f1",
"sha256": "0159d395d3236d038616de1bb2b1002ce1f2835f3715eaafb3f3ed8ca56fedad"
},
"downloads": -1,
"filename": "jax_scipy_spatial-0.2.5.tar.gz",
"has_sig": false,
"md5_digest": "cca0aabe2f78ce464b6b93ab77ac41f1",
"packagetype": "sdist",
"python_version": "source",
"requires_python": null,
"size": 16355,
"upload_time": "2024-11-27T16:26:13",
"upload_time_iso_8601": "2024-11-27T16:26:13.754329Z",
"url": "https://files.pythonhosted.org/packages/cc/eb/592cae2b7d5af29ab1113233bd0c8adb62e9bf6499884745140b422ab3d5/jax_scipy_spatial-0.2.5.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-11-27 16:26:13",
"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
