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.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.
Contributing
To run unit tests on your local machine:
tox
Project details
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.3.tar.gz
(17.2 kB
view hashes)
Built Distribution
Close
Hashes for jax_scipy_spatial-0.2.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0a8bf72aff329faed8c2c4de9292157f5679f01fab4ccf44fe52671e83d86449 |
|
MD5 | c41ac5290085c0070b068e63d90d5d0e |
|
BLAKE2b-256 | 23bd781ae308c222f5d4d9eaf885ff66633f7e73579fdff8b1185f52d2809885 |