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.1.tar.gz
(17.3 kB
view hashes)
Built Distribution
Close
Hashes for jax_scipy_spatial-0.2.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b829c8d9c2289ad49cffe92b8f0d73432a949728a00848b6973e64b8621f9231 |
|
MD5 | be421747d106dc461af90e8464e71b15 |
|
BLAKE2b-256 | e2edad98a909fb075a582be7bd59d2a3d7fac7d1367c10e65cd77322b091e7a4 |