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.6.tar.gz
(16.0 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.6.tar.gz.
File metadata
- Download URL: jax_scipy_spatial-0.2.6.tar.gz
- Upload date:
- Size: 16.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
3795905bf6a8e6a7795432827547a23d4d753ecccdfddda5693cf9e011961948
|
|
| MD5 |
b3c6ec94dbe4cb511ee1849433c17bb6
|
|
| BLAKE2b-256 |
5c8e268d3feb9026ca0d03f2e702a854216ab75f79b9abe36ed0357df0009c3e
|
File details
Details for the file jax_scipy_spatial-0.2.6-py3-none-any.whl.
File metadata
- Download URL: jax_scipy_spatial-0.2.6-py3-none-any.whl
- Upload date:
- Size: 13.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c099c3a1bde6c3577ad689c4a2da3d690a1963dc48911127b1e444299e5a87eb
|
|
| MD5 |
daba976d1822f2cac22c697d7d4f6f77
|
|
| BLAKE2b-256 |
551692ca2c113f756af0e18ecdbad6c8bccd5f7177de79d3e83ce02f48611aec
|
