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.0.tar.gz
(17.3 kB
view hashes)
Built Distribution
Close
Hashes for jax_scipy_spatial-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 37ee5794b8b5b32f446d0b7d2c4db6ca9cfe36375d27f899d37f0d87968c1e83 |
|
MD5 | a8549a2bf5e850dae21b626b8285b93d |
|
BLAKE2b-256 | a61d2786ac76767f893b0f73649eeccb279897ac960cc72bce09918cb6a7938e |