LouisDesdoigts / dLuxView external linksLinks
Differentiable optical models as parameterised neural networks in Jax using Zodiax
☆65Dec 1, 2025Updated 2 months ago
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☆24Dec 16, 2025Updated last month
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆115Feb 3, 2026Updated last week
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆19Jan 23, 2026Updated 3 weeks ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆43Nov 27, 2025Updated 2 months ago
- ☆15Aug 28, 2023Updated 2 years ago
- Shape Measurement Pipeline☆17Updated this week
- JAX-powered Cosmological Particle-Mesh N-body Solver☆58Dec 15, 2025Updated last month
- c++ backed python optical raytracer☆25Feb 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 3 months ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 4 months ago
- Fast Linearized Coronagraph Optimizer (FALCO) in Python 3☆12Jan 8, 2026Updated last month
- poppy derivatives☆22Aug 13, 2021Updated 4 years ago
- Poke (pronounced poh-keh) is a Polarization Ray Tracing and Gaussian Beamlet module for Python☆56May 5, 2025Updated 9 months ago
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 3 years ago
- simple simulations for testing weak lensing shear measurement☆12Oct 31, 2025Updated 3 months ago
- Data-driven wavefront-based PSF modelling framework.☆20Updated 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"☆13Mar 18, 2024Updated last year
- XLuminA, a highly-efficient, auto-differentiating discovery framework for super-resolution microscopy.☆55May 19, 2025Updated 8 months ago
- Special functions in Jax☆16Mar 1, 2025Updated 11 months ago
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆52Updated this week
- Interpolation and function approximation with JAX☆248Feb 2, 2026Updated last week
- Differentiable Gravitational Waveforms with JAX☆71Jul 28, 2025Updated 6 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆541Updated this week
- A framework for performing optical propagation simulations, meant for high contrast imaging, in Python.☆129Feb 7, 2026Updated last week
- ☆33Jan 12, 2026Updated last month
- A differentiable cosmology library in JAX☆223Jun 27, 2025Updated 7 months ago
- Fourier Modal Method for multilayer metamaterials. Mirror of https://gitlab.com/nannos/nannos☆20Sep 5, 2024Updated last year
- A fast, accurate estimator for small shear distortion.☆15Oct 10, 2024Updated last year
- 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.☆32Feb 8, 2026Updated last week
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last week
- S2FFT: Differentiable and accelerated spherical transforms☆215Updated this week
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆17Mar 5, 2024Updated last year
- pjz is JAX and fdtd-z -- scalable photonic simulation and optimization☆21Mar 27, 2024Updated last year