JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro
☆16Feb 25, 2025Updated last year
Alternatives and similar repositories for JaxTutos
Users that are interested in JaxTutos are comparing it to the libraries listed below
Sorting:
- Modern normalizing flows in Python. Simple to use and easily extensible.☆12Feb 11, 2026Updated 3 weeks ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆12Jul 17, 2025Updated 7 months ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- Autoencoders for galaxies☆11Jul 29, 2021Updated 4 years ago
- Differentiable Cosmological Forward Model☆90Feb 26, 2026Updated last week
- Interactive notebooks for graduate course in galaxies.☆16Jun 14, 2024Updated last year
- Library for meta-detection, combining detection and metacalibration☆13Jan 23, 2026Updated last month
- Save All Correlations and Covariances☆15Dec 1, 2025Updated 3 months ago
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- Shape Measurement Pipeline☆17Feb 13, 2026Updated 3 weeks ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆25May 27, 2025Updated 9 months ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆27Oct 13, 2022Updated 3 years ago
- PRObabilistic Value-Added Bright Galaxy Survey (PROVABGS)☆18Aug 12, 2024Updated last year
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆17Mar 5, 2024Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Feb 12, 2026Updated 3 weeks ago
- differentiable boltzmann code☆44Feb 6, 2025Updated last year
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- MiniAero Unstructured Finite Volume Compressible Navier-Stokes Mini-App☆22May 9, 2024Updated last year
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆20Updated this week
- Differentiable strong lens modelling on CPUs and GPUs☆28Feb 25, 2026Updated last week
- Tools for mesh adaptation☆23Dec 7, 2023Updated 2 years ago
- Curated list of astronomy/astrophysics code packages using JAX☆52Feb 18, 2026Updated 2 weeks ago
- Header-only C++-Library for grid file I/O☆22Jan 29, 2026Updated last month
- ☆28Updated this week
- This aims to be an wrapper to C-MPI3 for C++, using the principles of simplicity, STL, RAII and Boost and enforcing type-safety. This i…☆23Oct 11, 2024Updated last year
- This repository will hold a series of Python Tutorials dedicated to astronomers that are interested in learning Python.☆12Nov 26, 2019Updated 6 years ago
- Unstructured mesh hydrodynamics for advanced architectures☆23Aug 16, 2023Updated 2 years ago
- Particle Mesh in Python☆26Sep 22, 2025Updated 5 months ago
- a wavelet-based multifractal image analysis tool implementing the WTMM (Wavelet Transform Modulus Maxima) method.☆11Feb 1, 2020Updated 6 years ago
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Updated this week
- Modern cosmological inference framework used by the Dark Energy Spectroscopic Instrument (DESI) experiment.☆33Updated this week
- A unified pseudo-Cl framework☆65Feb 24, 2026Updated last week
- A Jupyter/python-based introduction to writing a finite-volume hydro code☆31Jul 26, 2019Updated 6 years ago
- Examples for usage of the Core Cosmology Library☆32Apr 20, 2025Updated 10 months ago
- Code for Gaussian Score Matching Variational Inference☆35Feb 23, 2025Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆32May 12, 2023Updated 2 years ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- A differentiable cosmology library in JAX☆225Jun 27, 2025Updated 8 months ago