LouisDesdoigts / dLuxLinks
Differentiable optical models as parameterised neural networks in Jax using Zodiax
☆59Updated 2 weeks ago
Alternatives and similar repositories for dLux
Users that are interested in dLux are comparing it to the libraries listed below
Sorting:
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆100Updated last week
- c++ backed python optical raytracer☆25Updated 7 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated 4 months ago
- Data-driven wavefront-based PSF modelling framework.☆19Updated this week
- poppy derivatives☆22Updated 4 years ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆51Updated last week
- Unitful Quantities in JAX☆43Updated 2 weeks ago
- Object-oriented Jax framework extending Equinox for scientific programming☆18Updated 7 months ago
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆62Updated 2 weeks ago
- Differentiable Gravitational Waveforms with JAX☆61Updated last month
- JAX bindings for the NVIDIA cuDecomp library☆44Updated this week
- A Jax wrapper for cudaKDTree☆10Updated 2 years ago
- A differentiable cosmology library in JAX☆213Updated 2 months ago
- Differentiable Cosmological Forward Model☆85Updated this week
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆15Updated 6 months ago
- Curated list of astronomy/astrophysics code packages using JAX☆45Updated 9 months ago
- hankl is a lightweight Python implementation of the FFTLog algorithm for Cosmology☆26Updated 2 years ago
- Numerical quadrature with JAX☆69Updated this week
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆25Updated last year
- A JAX package for weak gravitational lensing☆35Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Updated last year
- Educational materials for topics related to CosmoStat activities.☆39Updated 2 years ago
- Gravitational-wave data analysis tools in Jax☆74Updated last month
- Coverage tests to check the quality of your posterior estimators.☆32Updated last year
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆31Updated 3 weeks ago
- Astronomical time series analysis with JAX☆65Updated this week
- Poke (pronounced poh-keh) is a Polarization Ray Tracing and Gaussian Beamlet module for Python☆45Updated 4 months ago
- A webpage for the scattering transform. To calculate, you just need the folder called "scattering".☆60Updated 11 months ago
- Coordinates in JAX☆25Updated 2 weeks ago
- nessai: Nested Sampling with Artificial Intelligence☆41Updated last week