Differentiable Universe Initiative

Differentiable Universe Initiative's repositories

jax_cosmo

A differentiable cosmology library in JAX

Language:PythonMIT156 18 58

flowpm

Particle Mesh simulation in TensorFlow

Language:Jupyter NotebookMIT89 13 40

jaxDecomp

JAX bindings for the NVIDIA cuDecomp library

Language:PythonMIT2003

JaxPM

JAX-powered Cosmological Particle-Mesh N-body Solver

Language:Jupyter NotebookMIT18 2 5

jax-cosmo-paper

Paper introducing jax-cosmo

Language:Jupyter NotebookMIT14 5 9

GalFlow

A TensorFlow reimplementation of GalSim

Language:Jupyter NotebookMIT10 6 9

DHOD

Differentiable Halo Occupation Distributions

Language:Jupyter NotebookMIT8 5 16

sbi_lens

A python package for Weak Lensing Implicit Inference with Differentiable Simulator

Language:PythonMIT8 1 17

Initiative

Host repository for the initiative

MIT3 6 6

IDRIS-hackathon

Repository for hosting material and discussions for the 2021 IDRIS GPU hackathon

Language:PythonMIT2 4 5

jaxpm-paper

Repo for paper on Lagragian Neural ODEs for fast N-body simulations

Language:Jupyter NotebookMIT202

addons

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons

Language:PythonApache-2.002 1

Baryonator

Project for particle-based, minimally parametric, baryonification of fast PM simulations.

Language:Jupyter NotebookMIT02 2

Baryonification

Project for implementing baryonification techniques

MIT010

horovod

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

Language:PythonNOASSERTION03 3

mesh

Mesh TensorFlow: Model Parallelism Made Easier

Language:PythonApache-2.003 5

talks

Presentations maintained by the Initiative

MIT010

zelos

Greek god of emulation

MIT020