Skip to content

@DifferentiableUniverseInitiative Differentiable Universe Initiative

Change the repository type filter

All

    Repositories list

    18 repositories

    • jaxDecomp

      Public
      JAX bindings for the NVIDIA cuDecomp library
      hpcjaxxlacuda
      Python
      •
      MIT License
      •1•24•2•1•Updated Nov 29, 2024Nov 29, 2024

    • JaxPM

      Public
      JAX-powered Cosmological Particle-Mesh N-body Solver
      Jupyter Notebook
      •
      MIT License
      •14•29•2•2•Updated Nov 14, 2024Nov 14, 2024

    • jax_cosmo

      Public
      A differentiable cosmology library in JAX
      Python
      •
      MIT License
      •37•180•33•19•Updated Jul 31, 2024Jul 31, 2024

    • sbi_lens

      Public
      A python package for Weak Lensing Implicit Inference with Differentiable Simulator
      Python
      •
      MIT License
      •1•12•0•1•Updated Mar 5, 2024Mar 5, 2024

    • 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
      •1•2•1•0•Updated Jul 2, 2023Jul 2, 2023

    • jax-cosmo-paper

      Public
      Paper introducing jax-cosmo
      Jupyter Notebook
      •
      MIT License
      •7•14•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•90•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
      •4•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
      •611•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