Skip to content

@DifferentiableUniverseInitiative Differentiable Universe Initiative

Change the repository type filter

All

    Repositories list

    18 repositories

    • JaxPM

      Public
      JAX-powered Cosmological Particle-Mesh N-body Solver
      Jupyter Notebook
      •
      MIT License
      •24•62•1•3•Updated Mar 31, 2026Mar 31, 2026

    • sbi_lens

      Public
      A python package for Weak Lensing Implicit Inference with Differentiable Simulator
      Python
      •
      MIT License
      •5•17•0•2•Updated Mar 23, 2026Mar 23, 2026

    • jaxDecomp

      Public
      JAX reimplementation and bindings for the NVIDIA cuDecomp library
      hpccudajax
      hpccudajaxxla
      Python
      •
      MIT License
      •1•55•2•0•Updated Mar 23, 2026Mar 23, 2026

    • jax_cosmo

      Public
      A differentiable cosmology library in JAX
      Python
      •
      MIT License
      •47•233•38•21•Updated Jun 27, 2025Jun 27, 2025

    • Baryonator

      Public
      Project for particle-based, minimally parametric, baryonification of fast PM simulations.
      Jupyter Notebook
      •
      MIT License
      •0•0•2•0•Updated Jan 24, 2024Jan 24, 2024

    • jaxpm-paper

      Public
      Repo for paper on Lagragian Neural ODEs for fast N-body simulations
      Jupyter Notebook
      •
      MIT License
      •0•2•1•0•Updated Jul 2, 2023Jul 2, 2023

    • jax-cosmo-paper

      Public
      Paper introducing jax-cosmo
      Jupyter Notebook
      •
      MIT License
      •6•13•1•0•Updated Apr 27, 2023Apr 27, 2023

    • Initiative

      Public
      Host repository for the initiative
      MIT License
      •0•3•5•3•Updated Apr 12, 2023Apr 12, 2023

    • flowpm

      Public
      Particle Mesh simulation in TensorFlow
      Jupyter Notebook
      •
      MIT License
      •20•92•14•4•Updated Dec 10, 2022Dec 10, 2022

    • DHOD

      Public
      Differentiable Halo Occupation Distributions
      Jupyter Notebook
      •
      MIT License
      •2•8•5•1•Updated Nov 3, 2022Nov 3, 2022

    • GalFlow

      Public
      A TensorFlow reimplementation of GalSim
      Jupyter Notebook
      •
      MIT License
      •5•10•6•1•Updated Mar 8, 2022Mar 8, 2022

    • mesh

      Public
      Mesh TensorFlow: Model Parallelism Made Easier
      Python
      •
      Apache License 2.0
      •255•0•2•1•Updated Jan 6, 2022Jan 6, 2022

    • addons

      Public
      Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
      Python
      •
      Apache License 2.0
      •620•0•1•0•Updated Jun 20, 2021Jun 20, 2021

    • IDRIS-hackathon

      Public
      Repository for hosting material and discussions for the 2021 IDRIS GPU hackathon
      Python
      •
      MIT License
      •0•2•3•0•Updated May 27, 2021May 27, 2021

    • horovod

      Public
      Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
      Python
      •
      Other
      •2.2k•0•1•2•Updated May 25, 2021May 25, 2021

    • zelos

      Public
      Greek god of emulation
      MIT License
      •0•0•0•0•Updated May 29, 2020May 29, 2020

    • Baryonification

      Public
      Project for implementing baryonification techniques
      MIT License
      •1•0•0•1•Updated Mar 20, 2020Mar 20, 2020

    • talks

      Public
      Presentations maintained by the Initiative
      MIT License
      •0•0•0•0•Updated Dec 4, 2019Dec 4, 2019
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.