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.2.tar.gz
(17.2 kB
view hashes)
Built Distribution
Close
Hashes for jax_scipy_spatial-0.2.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e652664a704c87e253290c9640eceb7eb88efe76a54d71ed906b1340ad00e745 |
|
MD5 | a93585bf5a01ade214dfce348df2a2cf |
|
BLAKE2b-256 | 07cfe0b01d2cbc0b6de7c58d182575865f8e317f235e392eaac73b10a188c41b |