jax-scipy-spatial 0.2.4

Creator: railscoder56

Last updated:

Add to Cart

Description:

jaxscipyspatial 0.2.4

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

License

For personal and professional use. You cannot resell or redistribute these repositories in their original state.

Customer Reviews

There are no reviews.