astrojax

Astrojax - Accelerated astrodynamics in JAX

Astrodynamics written in JAX for massively parallel simulation.

The goal of this project is to provide a high-performance astrodynamics library that can be used for research and education. The library is built on top of JAX, which allows for automatic differentiation and GPU/TPU acceleration. The goal is to provide a proof-of-concept implementation of common astrodynamics algorithms that can be used as a starting point for further development. It is not intended to be a full-featured, high-accuracy astrodynamics library (at least not yet).

Install

pip install astrojax
# or
uv add astrojax

If you have a GPU and want to take advantage of JAX's acceleration can then install the package with the appropriate JAX extras:

pip install astrojax[cuda13]
# or
uv add astrojax[cuda13]

Astrojax supports CUDA 12 and CUDA 13.

Quickstart

This project uses just as a command runner and uv for Python package management.

# Install just (macOS)
brew install just

# Install uv
curl -LsSf https://astral.sh/uv/install.sh | sh

# Clone and install everything (dev deps + extras)
just install

Development

Every recipe has an equivalent raw command you can run directly.

Task	`just` recipe	Raw command
Install dev deps + extras	`just install`	`uv sync --dev --extra extras`
Run tests	`just test`	`uv run pytest tests/ -v`
Test with coverage	`just test-cov`	`uv run pytest --cov=astrojax --cov-report=term-missing`
Format code	`just fmt`	`uv run ruff format`
Lint (auto-fix)	`just lint`	`uv run ruff check --fix`
Type check	`just typecheck`	`uvx ty check`
All quality checks	`just check`	Runs fmt + lint + typecheck
Build docs	`just docs-build`	`uv run zensical build --clean`
Serve docs	`just docs-serve`	`uv run zensical serve --clean`

License

The code in this repository is licensed under the MIT License. See LICENSE for details.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

astrojax-0.8.0.tar.gz (4.8 MB view details)

Uploaded Jul 31, 2026 Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

astrojax-0.8.0-py3-none-any.whl (4.9 MB view details)

Uploaded Jul 31, 2026 Python 3

File details

Details for the file astrojax-0.8.0.tar.gz.

File metadata

Download URL: astrojax-0.8.0.tar.gz
Upload date: Jul 31, 2026
Size: 4.8 MB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for astrojax-0.8.0.tar.gz
Algorithm	Hash digest
SHA256	`4ac05283df00e519be16e67f10079abd83db5e6491ac767055b932a15b4577e1`
MD5	`4fce146124f2a72dfe1379c405c5813c`
BLAKE2b-256	`d3960863a1ed27e2269b87dd3df813a2193069d7c2340f7492260dc717a3ecc1`

See more details on using hashes here.

Provenance

The following attestation bundles were made for astrojax-0.8.0.tar.gz:

Publisher: release.yml on duncaneddy/astrojax

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: astrojax-0.8.0.tar.gz
- Subject digest: 4ac05283df00e519be16e67f10079abd83db5e6491ac767055b932a15b4577e1
- Sigstore transparency entry: 2303691330
- Sigstore integration time: Jul 31, 2026
Source repository:
- Permalink: duncaneddy/astrojax@928c35a704276205a827b427bee5ae833ed1d7b4
- Branch / Tag: refs/tags/v0.8.0
- Owner: https://github.com/duncaneddy
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: release.yml@928c35a704276205a827b427bee5ae833ed1d7b4
- Trigger Event: push