JAX-GalSim

JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.

Disclaimer: This project is still in an early development phase, please use the reference GalSim implementation for any scientific applications.

You can find the most up-to-date documentation for the project here.

Objective and Design

The goal of this library is to reimplement GalSim functionalities in pure JAX to allow for automatic differentiation, GPU acceleration, and batched computations.

Guiding Principles

Strive to be a drop-in replacement for GalSim, i.e. provide a close match to the GalSim API.
Each function/feature will be tested against the reference GalSim implementation.
This package will aim to be a subset of GalSim (i.e. only contains functions with a reference GalSim implementation).
Implementations should be easy to read and understand.
Code should be pip-installable on any machine, no compilation required.
Any notable differences between the JAX and reference implementations will be clearly documented.

Notable Differences

JAX arrays are immutable. Thus, in-place operations on images are not possible and a new image is returned instead. Also, the RNG classes cannot fill arrays and instead return new arrays.
JAX arrays do not support all the kinds of views that numpy arrays support. This means that some operations that work in GalSim may not work in JAX-GalSim (e.g., real views of complex images).
JAX-GalSim uses a different random number generator than GalSim. This leads to different results in terms of both the generated random numbers and in terms of which RNGs have stable discarding.

Contributing

Everyone can contribute to this project, please refer to the CONTRIBUTING.md document for details.

In short, to interact with the project you can:

Ask or Answer questions on the Discussions Q&A page
Report a bug by opening a GitHub issue
Open a GitHub issue or Discussions to ask for feedback on a planned contribution.
Submit a Pull Request to contribute to the code.

Issues marked with contributions welcome or good first issue are particularly good places to start. These are great ways to learn more about the inner workings of GalSim and how to code in JAX.

Current GalSim API Coverage

See the corresponding documentation page for a list of what is currently implemented.

Download files

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

Source Distribution

jax_galsim-2026.7.0.tar.gz (417.6 kB view details)

Uploaded Jul 23, 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.

jax_galsim-2026.7.0-py3-none-any.whl (189.1 kB view details)

Uploaded Jul 23, 2026 Python 3

File details

Details for the file jax_galsim-2026.7.0.tar.gz.

File metadata

Download URL: jax_galsim-2026.7.0.tar.gz
Upload date: Jul 23, 2026
Size: 417.6 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for jax_galsim-2026.7.0.tar.gz
Algorithm	Hash digest
SHA256	`38491b94a6482314ee94cdd88d73a03461193656c346b1cbc1ca496f18990a0b`
MD5	`92fbc1bb5e296ead793a13f4289972d9`
BLAKE2b-256	`27901c5cee20f2f6dee0f291cc3b88f0d932276abd632883cc164b0eaabdc952`

See more details on using hashes here.

Provenance

The following attestation bundles were made for jax_galsim-2026.7.0.tar.gz:

Publisher: release.yml on GalSim-developers/JAX-GalSim

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: jax_galsim-2026.7.0.tar.gz
- Subject digest: 38491b94a6482314ee94cdd88d73a03461193656c346b1cbc1ca496f18990a0b
- Sigstore transparency entry: 2228688015
- Sigstore integration time: Jul 23, 2026
Source repository:
- Permalink: GalSim-developers/JAX-GalSim@a17b4f1883b7a06e0d19d1857b648a1676a50360
- Branch / Tag: refs/tags/v2026.7.0
- Owner: https://github.com/GalSim-developers
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: release.yml@a17b4f1883b7a06e0d19d1857b648a1676a50360
- Trigger Event: release