Differentiable Universe Initiative

JaxPM Public
JAX-powered Cosmological Particle-Mesh N-body Solver

DifferentiableUniverseInitiative/JaxPM’s past year of commit activity

Jupyter Notebook 59 MIT 23 1 1 Updated Dec 15, 2025
jaxDecomp Public
JAX bindings for the NVIDIA cuDecomp library

DifferentiableUniverseInitiative/jaxDecomp’s past year of commit activity

Python 50 MIT 1 3 0 Updated Oct 28, 2025
jax_cosmo Public
A differentiable cosmology library in JAX

DifferentiableUniverseInitiative/jax_cosmo’s past year of commit activity

Python 220 MIT 45 38 (3 issues need help) 18 Updated Jun 27, 2025
sbi_lens Public
A python package for Weak Lensing Implicit Inference with Differentiable Simulator

DifferentiableUniverseInitiative/sbi_lens’s past year of commit activity

Python 17 MIT 3 0 1 Updated Mar 5, 2024
Baryonator Public
Project for particle-based, minimally parametric, baryonification of fast PM simulations.

DifferentiableUniverseInitiative/Baryonator’s past year of commit activity

Jupyter Notebook 0 MIT 0 2 0 Updated Jan 24, 2024
jaxpm-paper Public
Repo for paper on Lagragian Neural ODEs for fast N-body simulations

DifferentiableUniverseInitiative/jaxpm-paper’s past year of commit activity

Jupyter Notebook 2 MIT 1 1 0 Updated Jul 2, 2023
jax-cosmo-paper Public
Paper introducing jax-cosmo

DifferentiableUniverseInitiative/jax-cosmo-paper’s past year of commit activity

Jupyter Notebook 13 MIT 6 1 0 Updated Apr 27, 2023
Initiative Public
Host repository for the initiative

DifferentiableUniverseInitiative/Initiative’s past year of commit activity

3 MIT 0 5 3 Updated Apr 12, 2023
flowpm Public
Particle Mesh simulation in TensorFlow

DifferentiableUniverseInitiative/flowpm’s past year of commit activity

Jupyter Notebook 91 MIT 20 14 (1 issue needs help) 4 Updated Dec 10, 2022
DHOD Public
Differentiable Halo Occupation Distributions

DifferentiableUniverseInitiative/DHOD’s past year of commit activity

Jupyter Notebook 8 MIT 2 5 1 Updated Nov 3, 2022

View all repositories

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Differentiable Universe Initiative

Pinned Loading

Repositories

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

People

Top languages

Uh oh!

Most used topics

Uh oh!

Search code, repositories, users, issues, pull requests...

Pinned Loading

Repositories

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

People

Top languages

Uh oh!

Most used topics

Uh oh!