Software

Core developer · Coordination Committee · Strategic Planning

The community core package for astronomy in Python. The collaboration received the 2025 AAS Lancelot M. Berkeley Prize, a 2023 IOP Publishing Top Cited Paper Award and the 2022 ADASS software prize.

Orbit integration, potentials and stream generation — GPU-accelerated and fully differentiable.

Multiple dispatch for Python, with type-hint-driven method resolution — the dispatch layer under quax and unxt.

Vectors, frames and transformations — differentiable, and unit-aware via unxt.

Unit-aware quantities that survive jit, grad and vmap.

Custom array-ish types that work with JAX primitives — the substrate the rest of the stack builds on.

A standard interface for cosmology libraries

Extras for Diffrax

Interoperability between galactic dynamics libraries

Check whether a type hint is Annotated

There's more to JAX

Constrain macroscopic dark matter with lightning

Multivariate Gaussian KDE

Construct checks for optional dependencies

Paths through phase space

Dispatch over plotting backends

Building blocks for quax classes

Pre-quaxed libraries

scipy.special in JAX

Stellar stream tracks

dataclasses, for everything

Singletons for Equinox

A LaTeX preamble worth reusing

An immutable mapping, JAX-compatible

Efficiently get the index-0 element of any iterable

Other Open-Source Contributions44