Differentiable optical models as parameterised neural networks in Jax using Zodiax
☆70Mar 24, 2026Updated 3 weeks ago
Alternatives and similar repositories for dLux
Users that are interested in dLux are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Object-oriented Jax framework extending Equinox for scientific programming☆26Apr 8, 2026Updated last week
- An open-source carbon box model implementation built on JAX.☆11Apr 2, 2024Updated 2 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆131Apr 6, 2026Updated 2 weeks ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated 2 years ago
- Fast Linearized Coronagraph Optimizer (FALCO) in Python 3☆12Mar 25, 2026Updated 3 weeks ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- AMI-GO brrrrrrr☆25Updated this week
- ☆16Aug 28, 2023Updated 2 years ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆21Updated this week
- c++ backed python optical raytracer☆26Apr 7, 2026Updated last week
- Differentiable cosmological emulators: the JAX version of CosmoPower☆45Nov 27, 2025Updated 4 months ago
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 6 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆63Mar 31, 2026Updated 2 weeks ago
- Poke (pronounced poh-keh) is a Polarization Ray Tracing and Gaussian Beamlet module for Python☆60May 5, 2025Updated 11 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Shape Measurement Pipeline☆17Mar 31, 2026Updated 2 weeks ago
- poppy derivatives☆22Aug 13, 2021Updated 4 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆571Apr 13, 2026Updated last week
- Examples for photonic design and simulation problems using various tools and methods.☆12Mar 21, 2023Updated 3 years ago
- Interpolation and function approximation with JAX☆258Updated this week
- ☆35Mar 5, 2026Updated last month
- XLuminA, a highly-efficient, auto-differentiating discovery framework for super-resolution microscopy.☆57May 19, 2025Updated 11 months ago
- Fourier Modal Method for multilayer metamaterials. Mirror of https://gitlab.com/nannos/nannos☆20Sep 5, 2024Updated last year
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- simple simulations for testing weak lensing shear measurement☆13Updated this week
- Repository with the code of "HNPE: Leveraging Global Parameters for Neural Posterior Estimation"☆14Mar 18, 2024Updated 2 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 6 months ago
- A differentiable cosmology library in JAX☆233Jun 27, 2025Updated 9 months ago
- Data-driven wavefront-based PSF modelling framework.☆21Apr 13, 2026Updated last week
- A framework for performing optical propagation simulations, meant for high contrast imaging, in Python.☆135Mar 19, 2026Updated last month
- Fourier modal method with Jax☆141Feb 10, 2026Updated 2 months ago
- James Webb Space Telescope PSF simulation tool☆132Dec 2, 2025Updated 4 months ago
- pjz is JAX and fdtd-z -- scalable photonic simulation and optimization☆21Mar 27, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- S2FFT: Differentiable and accelerated spherical transforms☆223Updated this week
- Orthogonal polynomials with JAX☆28Apr 13, 2026Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆539Apr 5, 2026Updated 2 weeks ago
- Building blocks for Mueller calculus☆11Dec 1, 2025Updated 4 months ago
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆56Mar 23, 2026Updated 3 weeks ago
- physical optics: integrated modeling, phase retrieval, segmented systems, polynomials and fitting, sequential raytracing...☆331Feb 17, 2026Updated 2 months ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Updated this week