Differentiable optical models as parameterised neural networks in Jax using Zodiax
☆66Mar 6, 2026Updated this week
Alternatives and similar repositories for dLux
Users that are interested in dLux are comparing it to the libraries listed below
Sorting:
- Object-oriented Jax framework extending Equinox for scientific programming☆24Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆121Updated this week
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆20Feb 13, 2026Updated 3 weeks ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- ☆16Aug 28, 2023Updated 2 years ago
- Shape Measurement Pipeline☆17Feb 13, 2026Updated 3 weeks ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆59Feb 24, 2026Updated last week
- c++ backed python optical raytracer☆26Feb 12, 2025Updated last year
- An open-source carbon box model implementation built on JAX.☆11Apr 2, 2024Updated last year
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 4 months ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 5 months ago
- Fast Linearized Coronagraph Optimizer (FALCO) in Python 3☆12Feb 11, 2026Updated 3 weeks ago
- poppy derivatives☆22Aug 13, 2021Updated 4 years ago
- Poke (pronounced poh-keh) is a Polarization Ray Tracing and Gaussian Beamlet module for Python☆57May 5, 2025Updated 10 months ago
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 3 years ago
- simple simulations for testing weak lensing shear measurement☆12Oct 31, 2025Updated 4 months ago
- Data-driven wavefront-based PSF modelling framework.☆21Updated this week
- Examples for photonic design and simulation problems using various tools and methods.☆12Mar 21, 2023Updated 2 years ago
- Repository with the code of "HNPE: Leveraging Global Parameters for Neural Posterior Estimation"☆14Mar 18, 2024Updated last year
- XLuminA, a highly-efficient, auto-differentiating discovery framework for super-resolution microscopy.☆56May 19, 2025Updated 9 months ago
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- Special functions in Jax☆17Mar 1, 2025Updated last year
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆54Feb 12, 2026Updated 3 weeks ago
- Interpolation and function approximation with JAX☆250Feb 16, 2026Updated 2 weeks ago
- Differentiable Gravitational Waveforms with JAX☆71Jul 28, 2025Updated 7 months ago
- A framework for performing optical propagation simulations, meant for high contrast imaging, in Python.☆129Updated this week
- ☆34Feb 26, 2026Updated last week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 2 weeks ago
- A differentiable cosmology library in JAX☆225Jun 27, 2025Updated 8 months ago
- A fast, accurate estimator for small shear distortion.☆15Oct 10, 2024Updated last year
- Fourier Modal Method for multilayer metamaterials. Mirror of https://gitlab.com/nannos/nannos☆20Sep 5, 2024Updated last year
- Fourier modal method with Jax☆138Feb 10, 2026Updated 3 weeks ago
- Coverage tests to check the quality of your posterior estimators.☆33Aug 12, 2024Updated last year
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Updated this week
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- S2FFT: Differentiable and accelerated spherical transforms☆218Updated this week
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆17Mar 5, 2024Updated 2 years ago