Software
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
- jobovy/galpy2019Galactic Dynamics in python
- jobovy/gaia_tools2019Tools for working with the @ESAGaia data and related data sets
- jobovy/bovy_mcmc2019MCMC routines
- astropy/astroplan2020Observation planning package for astronomers – maintainer @bmorris3
- adrn/gala2020Galactic and gravitational dynamics in Python
- webbjj/clustertools2020A Python package with tools for analysing star clusters.
- GalOrrery/discO2020Don't panic at the disc 𝒪: Illuminating Orthogonal Expansions of the Disc.
- astropy/package-template2020Template for packages that use Astropy. Maintainer: @astrofrog
- skypyproject/skypy2020SkyPy: A package for modelling the Universe.
- astropy/sphinx-astropy2021Sphinx functionality for Astropy
- GalOrrery/sampleSCF2021
- astropy/astropy-healpix2021BSD-licensed HEALPix for Astropy - maintained by @astrofrog and @lpsinger
- GalOrrery/interpolated-coordinates2022Univariate Interpolations of Astropy Coordinates
- GalOrrery/stream_mapper-core2022
- cosmology-api/cosmology.api2023The Cosmology API standard
- cosmology-api/cosmology-compat2023Wrapper base classes.
- cosmology-api/cosmology.compat.astropy2023The Cosmology API conformant wrapper for Astropy Cosmology.
- cosmology-api/cosmology.compat.camb2023Cosmology API compatibility layer for CAMB
- GalOrrery/stream_mapper-pytorch2023
- GalOrrery/stream_mapper-visualization2023
- cosmology-api/cosmology.compat.class2023
- ntessore/positional_defaults2023Python package to set defaults for any positional-only parameter
- cosmology-api/cosmology2023In-development
- daft-dev/daft2023Render probabilistic graphical models using matplotlib
- data-apis/array-api2023RFC document, tooling and other content related to the array API standard
- stellarstreams/cats2023Community Atlas of Tidal Streams
- astropy/sphinx-automodapi2024Sphinx extension for generating API documentation
- jax-ml/jax2024Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
- patrick-kidger/equinox2024Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
- ilikecubesnstuff/commensurability2024Packages for calculating orbit commensurabilities.
- ilikecubesnstuff/pidgey2024Python interface for dynamics in galaxies, using Agama, Gala, and Galpy as backends.
- pyro-ppl/numpyro2024Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
- joshspeagle/brutus2024Modeling stellar photometry with "brute force" methods
- isochrones/isochrones2024Pythonic stellar model grid access; easy MCMC fitting of stellar properties
- astropy/quantity-2.02024Prototype of Quantity 2.0 (experimental)
- data-apis/array-api-typing2024Static typing support for the array API standard
- f0uriest/interpax2025Interpolation and function approximation with JAX
- xggs-dev/potamides2025Constrain gravitational potentials using stellar stream curvature analysis
- quantity-dev/metrology-apis2025Standardised Metrology APIs in Python.
- patrick-kidger/jaxtyping2025Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
- numpy/numpy2025The fundamental package for scientific computing with Python.
- patrick-kidger/wadler_lindig2025A Wadler–Lindig pretty printer for Python
- charlottemyers/galactoPINNs2026
- jorenham/optype2026Opinionated typing package for precise type hints in Python