Coordinate axes for scientific computing in JAX
Project description
Coordax - Coordinate Axes for scientific computing in JAX
Coordax makes it easy to associate array dimensions with coordinates in the context of scientific simulation codes written in JAX. This allows for efficient and expressive manipulation of data defined on structured grids, enabling operations like differentiation and interpolation with respect to physical coordinates.
Key features
- Supports computation on locally-positional axes via coordinate map (
cmap) - Manages coordinate objects that carry discretization details and custom methods
- Compatible with lossless conversion to and from
Xarrayrepresentation.
Coordax is particularly well-suited for scientific simulations where it is crucial to propagate discretization details and associated objects throughout the computation, such as Earth system modeling of fluid dynamics. For other use cases, we believe penzai.core.named_axes may be a better fit.
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 Distributions
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 coordax-0.1.1-py3-none-any.whl.
File metadata
- Download URL: coordax-0.1.1-py3-none-any.whl
- Upload date:
- Size: 52.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.11.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
83b68770c8ff444984b8144741c3ea500f7dbe5f0d25756bf946b7da005c1c5e
|
|
| MD5 |
b8ed41d55d83cf0edacd6552ec8b70d5
|
|
| BLAKE2b-256 |
a56c89a13b71ea2e7c40334edb2b7e040a95b161ce216343b29c793e1be69167
|
