This is an alpha release (v0.1.7) of jax-bandflux, a JAX-based implementation of SNCosmo's bandflux function, specifically for the SALT3-NIR model.
🚧 Early Development Notice 🚧
This is an early development version with limited functionality. Users should note:
SALT3 bandflux calculations are implemented
JAX-based computation enables automatic differentiation and GPU acceleration
The API is subject to change as the package matures
Current Limitations:
Not all SNCosmo features are implemented
Limited model support (SALT3-NIR only)
API may undergo significant changes
Tested only on SALT3-NIR
This package is primarily intended for researchers who need JAX-based differentiable bandflux calculations for SALT3 models.
Feedback and contributions are welcome!