Differentiable Boltzmann solver for cosmology in JAX
☆56May 3, 2026Updated 3 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).☆18Jun 10, 2026Updated last month
- Benchmarking field-level cosmological inference from galaxy surveys.☆15Jul 17, 2025Updated last year
- Code for moving DM in Nbody sims and painting baryons to Nbody☆21Updated this week
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 10 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 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.
- 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
- Comprehensive MCP framework for Handley Research Group - productivity tools, scientific computing, and lab administration☆18Aug 2, 2026Updated last week
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆21Jul 10, 2026Updated 3 weeks ago
- ☆28Jul 31, 2026Updated last week
- SO Likelihoods and Theories☆18Updated this week
- Differentiable strong lens modelling on CPUs and GPUs☆32Jul 8, 2026Updated last month
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- Cobaya-Cosmolike Architecture (From Cosmolike Org)☆28Jul 3, 2026Updated last month
- Differentiable cosmological emulators: the JAX version of CosmoPower☆48Nov 27, 2025Updated 8 months 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.
- Differentiable Likelihood for CMB Analysis☆31Jun 5, 2026Updated 2 months ago
- Pipeline to compute Nx2pt angular power spectra and their covariances.☆14Updated this week
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- Cosmological map inference with deep learning☆16Nov 3, 2021Updated 4 years ago
- Unleash the true power of scheduling☆39Jul 9, 2026Updated last month
- Differentiable Cosmological Forward Model☆93Jun 12, 2026Updated last month
- Agentic layer for ASTRA — dedicated skills, workflow execution, and HPC/container management for reproducible research☆18Jul 31, 2026Updated last week
- differentiable boltzmann code☆44Feb 6, 2025Updated last year
- S2FFT: Differentiable and accelerated spherical transforms☆228Updated this week
- 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.
- Pipeline elements for 3x2pt analysis (shear-shear, shear-density, density-density) for DC2☆23Updated this week
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago
- ☆14Aug 8, 2023Updated 3 years ago
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆19Mar 23, 2026Updated 4 months ago
- A Python library for performing galaxy cluster mass reconstruction from weak lensing observables☆32Updated this week
- Generator for Large Scale Structure☆54Aug 1, 2026Updated last week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆21Mar 10, 2024Updated 2 years ago
- Euclid Visible Instrument Python package. Includes a simulator and various analysis codes.☆11Mar 25, 2015Updated 11 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆137Updated this week
- 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.
- ☆19May 31, 2026Updated 2 months ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 3 years ago
- Python code to interface with halo catalogs and other Abacus N-body data products☆24Updated this week
- Lecture material used in my 8 hour long PhD course on "Numerical Methods for Partial Differential Equations"☆16May 21, 2025Updated last year
- OneCovariance code for KiDS to play☆19Jun 25, 2026Updated last month
- Differentiable Gravitational Waveforms with JAX☆73Jul 28, 2025Updated last year
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago