Owner

DifferentiableUniverseInitiative

11 indexed repositories · View on GitHub

jax_cosmo
A differentiable cosmology library in JAX

Python · 246 stars
flowpm
Particle Mesh simulation in TensorFlow

Jupyter Notebook · 93 stars
JaxPM
JAX-powered Cosmological Particle-Mesh N-body Solver

Jupyter Notebook · 72 stars
jaxDecomp
JAX reimplementation and bindings for the NVIDIA cuDecomp library

Python · 61 stars
jax-cosmo-paper
Paper introducing jax-cosmo

Jupyter Notebook · 13 stars
GalFlow
A TensorFlow reimplementation of GalSim

Jupyter Notebook · 10 stars
DHOD
Differentiable Halo Occupation Distributions

Jupyter Notebook · 8 stars
Initiative
Host repository for the initiative

3 stars
IDRIS-hackathon
Repository for hosting material and discussions for the 2021 IDRIS GPU hackathon

Python · 2 stars
jaxpm-paper
Repo for paper on Lagragian Neural ODEs for fast N-body simulations

Jupyter Notebook · 2 stars
Baryonator
Project for particle-based, minimally parametric, baryonification of fast PM simulations.

Jupyter Notebook · 0 stars

Beginner friendly Everything

Open All

Perfologs Open

bookkeeping

Difficulty 3/5 1-2 days Newbie friendliness 30/100

DifferentiableUniverseInitiative/jaxDecomp#127 · Opened Jul 9, 2026
[Experiment] Studying impact on painting kernels Open

Experiment

DifferentiableUniverseInitiative/JaxPM#60 · 1 comment · 1 assignee · Opened May 31, 2026
Few updates and fixes JPM edition Open

DifferentiableUniverseInitiative/JaxPM#59 · 1 assignee · Opened May 28, 2026
Few updates and fixes Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax_cosmo#149 · 1 comment · Opened May 27, 2026
Discussion] Integrating discrete thermodynamic pressure (Fuzzball evaporation) into parametric PM simulations Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

DifferentiableUniverseInitiative/Baryonator#3 · Opened Apr 29, 2026
CMB power spectrum status Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

DifferentiableUniverseInitiative/jax_cosmo#142 · Opened Nov 6, 2025
Deprecation Warning Open

Difficulty 2/5 1-3 hours Newbie friendliness 75/100

DifferentiableUniverseInitiative/jax_cosmo#141 · 1 comment · 5 reactions · Opened Oct 14, 2025
Problem with jax_cosmo caching system Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#140 · 4 comments · Opened Jul 6, 2025
Ensure Born Convergence in jaxpm matches glass Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

DifferentiableUniverseInitiative/JaxPM#47 · 1 comment · Opened Jul 5, 2025
Distance between two redshifts Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#134 · 5 comments · Opened Apr 9, 2025
Support for general n-z Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

DifferentiableUniverseInitiative/jax_cosmo#132 · Opened Feb 11, 2025
TODO list Open

Difficulty 5/5 Over a week Newbie friendliness 30/100

DifferentiableUniverseInitiative/jaxDecomp#59 · 1 comment · Opened Feb 6, 2025
Error in `background.radial_comoving_distance` ? Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#128 · 6 comments · Opened Sep 27, 2024
_endpoints not defined in jax_cosmo.scipy.interpolate Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#127 · 2 comments · Opened Aug 30, 2024
modified gravity model Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax_cosmo#122 · Opened Mar 24, 2024
interpol crash Open

Difficulty 3/5 1-2 days Newbie friendliness 42/100

DifferentiableUniverseInitiative/jax_cosmo#121 · 1 comment · Opened Feb 20, 2024
trying to port existing MCMC code to execute it on JAX GPU Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax-cosmo-paper#15 · Opened Jan 12, 2024
Support for differentiable arbitrary linear operators? Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax_cosmo#120 · Opened Dec 30, 2023
CMB Lensing? Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax_cosmo#119 · Opened Oct 3, 2023
[TypeError] Ensure children is a list before calling children.pop() for a cosmology with the growth index $\gamma$ Open

Difficulty 1/5 Under an hour Newbie friendliness 50/100

DifferentiableUniverseInitiative/jax_cosmo#113 · Opened Jun 28, 2023
Jaxification of halo table Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

DifferentiableUniverseInitiative/Baryonator#2 · Opened Jun 20, 2023
referencing Open

Difficulty 2/5 1-3 hours Newbie friendliness 48/100

DifferentiableUniverseInitiative/jax_cosmo#112 · Opened Jun 7, 2023
Bug in `sigmasqr` function in `power.py` Open

Difficulty 2/5 1-3 hours Newbie friendliness 42/100

DifferentiableUniverseInitiative/jax_cosmo#109 · 1 comment · Opened Feb 14, 2023
More flexible n(z) parameterizations Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#108 · Opened Jan 24, 2023
jax warnings Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#106 · 2 comments · Opened Jan 3, 2023
jax-cosmo paper authorship request and comments open until Jan. 11th Open

Difficulty 1/5 Under an hour Newbie friendliness 10/100

DifferentiableUniverseInitiative/jax_cosmo#103 · Opened Dec 21, 2022
Implement T and P in the NFW halo reparam Open

DifferentiableUniverseInitiative/Baryonator#1 · 1 assignee · Opened Nov 7, 2022
Tensorflow 1 is unsupported in Colab. Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/flowpm#103 · Opened Sep 3, 2022
CCL comparison : growth factor Open

Difficulty 1/5 Under an hour Newbie friendliness 45/100

DifferentiableUniverseInitiative/jax_cosmo#102 · Opened Aug 20, 2022
Actually only returns the signal Cell not noise+signal Open

documentation

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#99 · Opened Jun 30, 2022
CCL comparison Neff=0 Open

DifferentiableUniverseInitiative/jax_cosmo#97 · 1 assignee · Opened May 27, 2022
Halofit: fnu correction Open

Difficulty 3/5 1-2 days Newbie friendliness 48/100

DifferentiableUniverseInitiative/jax_cosmo#96 · 1 comment · Opened May 25, 2022
Worse result with the new interpolator Open

bug

DifferentiableUniverseInitiative/GalFlow#18 · 4 comments · 1 assignee · Opened May 20, 2022
PGD implementation in Jax Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

DifferentiableUniverseInitiative/jaxpm-paper#2 · 5 comments · Opened May 16, 2022
option to compute only autocorrelation cells Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

DifferentiableUniverseInitiative/jax_cosmo#95 · 1 comment · Opened Apr 25, 2022
radial_comoving_distance: number of steps too low. Open

Difficulty 1/5 Under an hour Newbie friendliness 55/100

DifferentiableUniverseInitiative/jax_cosmo#94 · Opened Apr 11, 2022
Migration integration: ClenshawCurtis Open

Difficulty 5/5 Over a week Newbie friendliness 30/100

DifferentiableUniverseInitiative/jax_cosmo#92 · 1 comment · Opened Apr 10, 2022
interp(x, xp, fp) Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/jax_cosmo#91 · 2 comments · Opened Apr 9, 2022
Neural ODEs for FLOWPM and ODE TF correction Open

bug

DifferentiableUniverseInitiative/flowpm#101 · 11 comments · 2 reactions · 1 assignee · Opened Mar 29, 2022
Integrating the Sympy-based Boltzmann solver from PyCosmo Open

Cosmology enhancement

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax_cosmo#87 · 5 comments · Opened Jan 29, 2022
Notes for improvements Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

DifferentiableUniverseInitiative/jax_cosmo#81 · 5 comments · Opened Jan 19, 2022
implement variable mu(z) computation Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

DifferentiableUniverseInitiative/jax_cosmo#80 · Opened Jan 6, 2022
FlowPM Blog Post Out of Date Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

DifferentiableUniverseInitiative/flowpm#96 · 3 comments · Opened Oct 8, 2021
Why not sampling all the things? HMC sampling of each individual galaxy activation Open

investigation

Difficulty 5/5 Over a week Newbie friendliness 25/100

DifferentiableUniverseInitiative/DHOD#37 · 8 comments · Opened Aug 6, 2021
New versions of TFP cause NaN gradients for RelaxedBernoulli near p=0 and p=1 Open

bug

DifferentiableUniverseInitiative/DHOD#34 · 3 comments · 1 assignee · Opened Aug 4, 2021
Parameters of PGD and optimal weighting/cic compensation Open

enhancement

DifferentiableUniverseInitiative/flowpm#89 · 11 comments · 2 assignees · Opened Jul 9, 2021
Fix Continuous Integration Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

DifferentiableUniverseInitiative/flowpm#87 · Opened Jul 1, 2021
Additional validation test for ray-tracing implementation Open

DifferentiableUniverseInitiative/flowpm#85 · 5 comments · 1 assignee · Opened Jun 25, 2021
Add generic tools for image resampling Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 35/100

DifferentiableUniverseInitiative/GalFlow#10 · 3 comments · Opened Jun 20, 2021
Port Weak Lensing code to Mesh TensorFlow Open

enhancement Mesh TensorFlow

DifferentiableUniverseInitiative/flowpm#84 · 4 comments · 1 assignee · Opened Jun 4, 2021

DifferentiableUniverseInitiative

Get new issues in your inbox