redback-jax documentation

Welcome to redback-jax, a lightweight JAX-only version of the redback electromagnetic transient analysis package.

Contents:

• Installation
  • Requirements
  • Installation from PyPI
  • Installation from Source
  • Development Installation
• Quick Start
  • Installation
  • Float32 mode (recommended for GPU)
  • Bolometric light curves
  • Fitting bolometric data
  • Spectra and photometry
  • Available models
  • Next steps
• Transient Analysis Tutorial
  • Basic Transient Usage
  • Multi-band Photometry
  • Model Fitting and Plotting
  • Working with JAX Arrays
  • Spectrum Analysis
  • Advanced Usage
  • Error Handling
  • Next Steps
• API Reference
  • Main Package
  • Models
  • Utilities
  • Inference
• Examples
  • Bolometric inference with BlackJAX
  • Photometry fitting with the inference API
  • Kilonova bolometric fitting
  • Shock-powered models
• Contributing
  • Development Setup
  • Running Tests
  • Code Style
  • Documentation
  • Submitting Changes
  • Guidelines

About redback-jax

Redback-JAX provides JAX-based implementations for electromagnetic transient modeling and Bayesian inference, focusing on performance and automatic differentiation capabilities. This package is designed to be a lightweight alternative to the full redback package, leveraging JAX for fast computation and gradient-based inference.

Features

• JAX-based models: Fast, differentiable implementations of electromagnetic transient models

• Bayesian inference: Integration with modern probabilistic programming libraries

• Automatic differentiation: Gradient-based optimization and sampling

• GPU/TPU support: Leverage JAX’s hardware acceleration capabilities

Indices and tables

• Index

• Module Index

• Search Page