Coordinates in JAX
Project description
coordinax
Coordinates in JAX
Coordinax enables calculations with coordinates in JAX. Built on Equinox and Quax.
Installation
pip install coordinax
Documentation
Coming soon. In the meantime, if you've used astropy.coordinates
, then
coordinax
should be fairly intuitive.
Quick example
import coordinax as cx
import jax.numpy as jnp
from unxt import Quantity
q = cx.CartesianPos3D(
x=Quantity(jnp.arange(0, 10.0), "km"),
y=Quantity(jnp.arange(5, 15.0), "km"),
z=Quantity(jnp.arange(10, 20.0), "km"),
)
print(q)
# <CartesianPos3D (x[km], y[km], z[km])
# [[ 0. 5. 10.]
# [ 1. 6. 11.]
# ...
# [ 8. 13. 18.]
# [ 9. 14. 19.]]>
q2 = cx.represent_as(q, cx.SphericalPos)
print(q2)
# <SphericalPos (r[km], theta[rad], phi[rad])
# [[11.18 0.464 1.571]
# [12.57 0.505 1.406]
# ...
# [23.601 0.703 1.019]
# [25.259 0.719 0.999]]>
p = cx.CartesianVel3D(
d_x=Quantity(jnp.arange(0, 10.0), "m/s"),
d_y=Quantity(jnp.arange(5, 15.0), "m/s"),
d_z=Quantity(jnp.arange(10, 20.0), "m/s"),
)
print(p)
# <CartesianVel3D (d_x[m / s], d_y[m / s], d_z[m / s])
# [[ 0. 5. 10.]
# [ 1. 6. 11.]
# ...
# [ 8. 13. 18.]
# [ 9. 14. 19.]]>
p2 = cx.represent_as(p, cx.SphericalVel, q)
print(p2)
# <SphericalVel (d_r[m / s], d_theta[m rad / (km s)], d_phi[m rad / (km s)])
# [[ 1.118e+01 -3.886e-16 0.000e+00]
# [ 1.257e+01 -1.110e-16 0.000e+00]
# ...
# [ 2.360e+01 0.000e+00 0.000e+00]
# [ 2.526e+01 -2.776e-16 0.000e+00]]>
Citation
If you found this library to be useful in academic work, then please cite.
Development
We welcome contributions!
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
coordinax-0.13.2.tar.gz
(156.7 kB
view details)
Built Distribution
coordinax-0.13.2-py3-none-any.whl
(123.1 kB
view details)
File details
Details for the file coordinax-0.13.2.tar.gz
.
File metadata
- Download URL: coordinax-0.13.2.tar.gz
- Upload date:
- Size: 156.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | f20ebde89810634aad3e3c7ae945c2ac02fe39eeac8f6b4fed161b4a5b359e6a |
|
MD5 | 4f854470de17d54cef5de82640058a52 |
|
BLAKE2b-256 | 4701dfbb1b2505c4bbd685f8427dddc80454d7742542a97c93b869f9054b6bd2 |
Provenance
File details
Details for the file coordinax-0.13.2-py3-none-any.whl
.
File metadata
- Download URL: coordinax-0.13.2-py3-none-any.whl
- Upload date:
- Size: 123.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5a70f80b8ba739115609ba9fc7f21571dbc0fb0bf1db79a627ae26d13108f862 |
|
MD5 | 6dce087faaaafdab0744c7dd33f94ae0 |
|
BLAKE2b-256 | 2e0eb66a5fa28bf605de6b6f1f90856f9f0c2a475f61f4a1088e91d17ed66f9f |