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
Release files for jax-scipy-spatial 0.2.6
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| jax_scipy_spatial-0.2.6.tar.gz | 16.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax_scipy_spatial-0.2.6-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 29.0 kB
Release files / jax_scipy_spatial-0.2.6.tar.gz
| Download URL | jax_scipy_spatial-0.2.6.tar.gz |
|---|---|
| Size | 16.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
3795905bf6a8e6a7795432827547a23d4d753ecccdfddda5693cf9e011961948
|
|
BLAKE2b-256 checksum How to use checksums |
5c8e268d3feb9026ca0d03f2e702a854216ab75f79b9abe36ed0357df0009c3e
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.12.3
|
Release files / jax_scipy_spatial-0.2.6-py3-none-any.whl
| Download URL | jax_scipy_spatial-0.2.6-py3-none-any.whl |
|---|---|
| Size | 13.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
c099c3a1bde6c3577ad689c4a2da3d690a1963dc48911127b1e444299e5a87eb
|
|
BLAKE2b-256 checksum How to use checksums |
551692ca2c113f756af0e18ecdbad6c8bccd5f7177de79d3e83ce02f48611aec
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.12.3
|