https://github.com/adonath/jax-gamma-ray-prototype

A prototype implementation for 3D gamma-ray analysis in Jax
https://github.com/adonath/jax-gamma-ray-prototype

Last synced: 19 days ago
JSON representation

A prototype implementation for 3D gamma-ray analysis in Jax

• Host: GitHub
• URL: https://github.com/adonath/jax-gamma-ray-prototype
• Owner: adonath
• Created: 2024-06-13T19:54:19.000Z (7 months ago)
• Default Branch: main
• Last Pushed: 2024-06-28T12:09:15.000Z (6 months ago)
• Last Synced: 2024-10-28T08:11:59.053Z (2 months ago)
• Language: Jupyter Notebook
• Size: 74.2 KB
• Stars: 1
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.MD

Awesome Lists containing this project

README

        
# Gamma-ray data analysis with Jax Prototype

This repo contains a prototype of a gamma-ray data analysis framework using Jax. The main goal is to asses the feasibility of using Jax as a backend to the (Gammapy)[https://gammapy.org/] library. It contains a set of benchmarks and tests to compare the performance of Jax with respect to the standard Gammapy analysis, which is based on Numpy and Scipy.

We would like to understand how performance scales with:

- The size of the dataset

- The complexity of the model (number of sources)

- Use of hardware accelerators (GPU, TPU)

- Use of gradients of the likelihood function

- Homogeneous vs heterogeneous datasets

## Setup 

To run the benchmarks in this repository you need to install the following dependencies:

```bash

mamba env create -f environment.yml

conda activate jax-gamma-ray-prototype

```

```bash

python prepare-test-datasets.py

```

```bash

python run-benchmarks.py

```