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.distancescipy.spatial.transform.Rotationscipy.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
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 details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file jax-scipy-spatial-0.2.3.tar.gz.
File metadata
- Download URL: jax-scipy-spatial-0.2.3.tar.gz
- Upload date:
- Size: 17.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.8.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
22b402ffa2ebd553315e8bf9c85dc1fdbd7282cb66389b4ba9ac6e11a36ae78b
|
|
| MD5 |
5859160e69631d5d3fa2bf9539eba417
|
|
| BLAKE2b-256 |
776fd05a728b797f841362a15c1cc1c572958b2ca3560855a33032d14405a15c
|
File details
Details for the file jax_scipy_spatial-0.2.3-py3-none-any.whl.
File metadata
- Download URL: jax_scipy_spatial-0.2.3-py3-none-any.whl
- Upload date:
- Size: 14.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.8.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0a8bf72aff329faed8c2c4de9292157f5679f01fab4ccf44fe52671e83d86449
|
|
| MD5 |
c41ac5290085c0070b068e63d90d5d0e
|
|
| BLAKE2b-256 |
23bd781ae308c222f5d4d9eaf885ff66633f7e73579fdff8b1185f52d2809885
|
