4 papers
JZ-Tree: GPU friendly neighbour search and friends-of-friends with dual tree walks in JAX plus CUDA
Jens Stücker, Oliver Hahn, Lukas Winkler +2
Algorithms based on spatial tree traversal are widely regarded as among the most efficient and flexible approaches for many problems in CPU-based high-performance computing (HPC).…
DISCO-DJ II: a differentiable particle-mesh code for cosmology
Florian List, Oliver Hahn, Thomas Flöss +1
The mildly non-linear regime of cosmic structure formation holds much of the information that upcoming large-scale structure surveys aim to exploit, making fast and accurate predic…
Neural Network Reconstruction of Non-Gaussian Initial Conditions from Dark Matter Halos
Jelte Bottema, Thomas Flöss, P. Daniel Meerburg
We develop a machine learning approach to reconstructing the cosmological initial conditions from late-time dark matter halo number density fields in redshift space, with the goal…
Towards detecting Primordial non-Gaussianity in the CMB using Spherical Convolutional Neural Networks
Jorik Melsen, Thomas Flöss, P. Daniel Meerburg
This paper explores a novel application of spherical convolutional neural networks (CNNs) to detect primordial non-Gaussianity in the cosmic microwave background (CMB), a key probe…