Fast GPU-Powered and Auto-Differentiable Forward Modeling of IFU Data Cubes
Fuente:
arXiv
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909424615948288 |
|---|---|
| author | Çakır, Ufuk Schaible, Anna Lena Buck, Tobias |
| author_facet | Çakır, Ufuk Schaible, Anna Lena Buck, Tobias |
| contents | We present RUBIX, a fully tested, well-documented, and modular Open Source tool developed in JAX, designed to forward model IFU cubes of galaxies from cosmological hydrodynamical simulations. The code automatically parallelizes computations across multiple GPUs, demonstrating performance improvements over state-of-the-art codes by a factor of 600. This optimization reduces compute times from hours to only seconds. RUBIX leverages JAX's auto-differentiation capabilities to enable not only forward modeling but also gradient computations through the entire pipeline paving the way for new methodological approaches such as e.g. gradient-based optimization of astrophysics model parameters. RUBIX is open-source and available on GitHub: https://github.com/ufuk-cakir/rubix. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2412_08265 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Fast GPU-Powered and Auto-Differentiable Forward Modeling of IFU Data Cubes Çakır, Ufuk Schaible, Anna Lena Buck, Tobias Instrumentation and Methods for Astrophysics Astrophysics of Galaxies Computational Physics Data Analysis, Statistics and Probability We present RUBIX, a fully tested, well-documented, and modular Open Source tool developed in JAX, designed to forward model IFU cubes of galaxies from cosmological hydrodynamical simulations. The code automatically parallelizes computations across multiple GPUs, demonstrating performance improvements over state-of-the-art codes by a factor of 600. This optimization reduces compute times from hours to only seconds. RUBIX leverages JAX's auto-differentiation capabilities to enable not only forward modeling but also gradient computations through the entire pipeline paving the way for new methodological approaches such as e.g. gradient-based optimization of astrophysics model parameters. RUBIX is open-source and available on GitHub: https://github.com/ufuk-cakir/rubix. |
| title | Fast GPU-Powered and Auto-Differentiable Forward Modeling of IFU Data Cubes |
| topic | Instrumentation and Methods for Astrophysics Astrophysics of Galaxies Computational Physics Data Analysis, Statistics and Probability |
| url | https://arxiv.org/abs/2412.08265 |