Skip to content
@DifferentiableUniverseInitiative

Differentiable Universe Initiative

Pinned Loading

  1. jax_cosmo jax_cosmo Public

    A differentiable cosmology library in JAX

    Python 247 50

  2. flowpm flowpm Public

    Particle Mesh simulation in TensorFlow

    Jupyter Notebook 93 20

  3. GalFlow GalFlow Public

    A TensorFlow reimplementation of GalSim

    Jupyter Notebook 10 5

  4. DHOD DHOD Public

    Differentiable Halo Occupation Distributions

    Jupyter Notebook 8 2

Repositories

Showing 10 of 19 repositories
  • jax-fli Public

    Forward-modeling and sampling on top of JAXPM + JAX-Decomp

    DifferentiableUniverseInitiative/jax-fli's past year of commit activity
    Python 2 MIT 0 4 5 Updated Oct 9, 2026
  • JaxPM Public

    JAX-powered Cosmological Particle-Mesh N-body Solver

    DifferentiableUniverseInitiative/JaxPM's past year of commit activity
    Python 72 MIT 26 2 1 Updated Sep 28, 2026
  • jaxDecomp Public

    JAX reimplementation and bindings for the NVIDIA cuDecomp library

    DifferentiableUniverseInitiative/jaxDecomp's past year of commit activity
    Python 63 MIT 3 2 0 Updated Sep 26, 2026
  • sbi_lens Public

    A python package for Weak Lensing Implicit Inference with Differentiable Simulator

    DifferentiableUniverseInitiative/sbi_lens's past year of commit activity
    Python 19 MIT 6 0 2 Updated Mar 23, 2026
  • jax_cosmo Public

    A differentiable cosmology library in JAX

    DifferentiableUniverseInitiative/jax_cosmo's past year of commit activity
    Python 247 MIT 50 39 (3 issues need help) 22 Updated Jun 27, 2025
  • 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 3 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 0 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 93 MIT 20 14 (1 issue needs help) 4 Updated Dec 10, 2022

Most used topics

Loading…