Scipy spatial API for JAX
Project description
jax-scipy-spatial
This package implements scipy.spatial API for JAX.
Currently the following items are implemented:
scipy.spatial.distancescipy.spatial.transform.Rotationscipy.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.
Contributing
To run unit tests on your local machine:
tox
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
jax_scipy_spatial-0.2.4.tar.gz
(16.3 kB
view details)
Built Distribution
File details
Details for the file jax_scipy_spatial-0.2.4.tar.gz.
File metadata
- Download URL: jax_scipy_spatial-0.2.4.tar.gz
- Upload date:
- Size: 16.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.8.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 718bd0ce37b50ff8a44d4a8d6a8ceefa11aa38b3f4639f23fb67f9b0130f7586 |
|
| MD5 | f99b793876c4921f2317ee632a69af0c |
|
| BLAKE2b-256 | 8541e22eb78c7d5519f2e0bf2d799a87dc273a7620c976f8b6066af1936df043 |
File details
Details for the file jax_scipy_spatial-0.2.4-py3-none-any.whl.
File metadata
- Download URL: jax_scipy_spatial-0.2.4-py3-none-any.whl
- Upload date:
- Size: 13.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.8.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 9e41ffd173cbd8f145859a86504833490dca010f70402a20f3265e4a8a7422c6 |
|
| MD5 | 19fd913d8ed86b4f17cb47d64bbdc733 |
|
| BLAKE2b-256 | 55c529fd515923afbd581f8dadfd8600b3ed73969bb07247353bc980d384c841 |
