Differentiable Boltzmann solver for cosmology in JAX
☆56May 3, 2026Updated 2 months ago
Alternatives and similar repositories for clax
Users that are interested in clax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Differentiable Boltzmann solver + one-loop EFTofLSS engine in JAX (mirrors CLASS and CLASS-PT).☆17Jun 10, 2026Updated last month
- SO Likelihoods and Theories☆18Updated this week
- Benchmarking field-level cosmological inference from galaxy surveys.☆14Jul 17, 2025Updated 11 months ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 9 months ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆25Jan 3, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Investigating the information content in the cosmic web through dark matter halo graphs. Arxiv paper: https://arxiv.org/abs/2207.05202☆13Jan 18, 2024Updated 2 years ago
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆21Updated this week
- ☆28Jul 7, 2026Updated last week
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- Differentiable Likelihood for CMB Analysis☆31Jun 5, 2026Updated last month
- Differentiable cosmological emulators: the JAX version of CosmoPower☆48Nov 27, 2025Updated 7 months ago
- Pipeline to compute Nx2pt angular power spectra and their covariances.☆14Jun 19, 2026Updated 3 weeks ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- ☆19Jun 25, 2026Updated 2 weeks ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Cobaya-Cosmolike Architecture (From Cosmolike Org)☆27Jul 3, 2026Updated last week
- Cosmological map inference with deep learning☆15Nov 3, 2021Updated 4 years ago
- Unleash the true power of scheduling☆39Updated this week
- Differentiable Cosmological Forward Model☆93Jun 12, 2026Updated last month
- MPI-Rockstar: a Hybrid MPI and OpenMP Parallel Implementation of the Rockstar Halo finder☆15Apr 27, 2026Updated 2 months ago
- differentiable boltzmann code☆44Feb 6, 2025Updated last year
- S2FFT: Differentiable and accelerated spherical transforms☆227Updated this week
- Euclid Visible Instrument Python package. Includes a simulator and various analysis codes.☆11Mar 25, 2015Updated 11 years ago
- Pipeline elements for 3x2pt analysis (shear-shear, shear-density, density-density) for DC2☆22Updated this week
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Agentic layer for ASTRA — dedicated skills, workflow execution, and HPC/container management for reproducible research☆18Updated this week
- Corner plots -- now with ads!☆23Mar 31, 2025Updated last year
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago
- ☆14Aug 8, 2023Updated 2 years ago
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆18Mar 23, 2026Updated 3 months ago
- A Python library for performing galaxy cluster mass reconstruction from weak lensing observables☆32May 13, 2026Updated 2 months ago
- Differentiable and GPU accelerated scattering covariance statistics on the sphere☆16May 27, 2025Updated last year
- A code for generating fake spectra from a cosmological simulation☆17Mar 26, 2026Updated 3 months ago
- ☆15May 31, 2026Updated last month
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated 2 years ago
- Implementation of various equivariant models in JAX☆19Apr 12, 2024Updated 2 years ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆16Updated this week
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 4 years ago
- Gravitational-wave astronomy in JAX☆15Updated this week
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 3 years ago
- Differentiable models of halo and galaxy formation history☆11Jul 1, 2026Updated last week