stella

stella is an open-source Python framework for identifying stellar flares in TESS two-minute cadence data using convolutional neural networks (Keras 3 on JAX or PyTorch).

• Backends: set KERAS_BACKEND to jax (default) or torch.
• Packaged models: from stella import models as sm; sm.models.
• Quickstart notebooks live under Getting Started.

Getting started

export KERAS_BACKEND=jax  # or torch
pip install -e .[dev]

Citations

• Feinstein, Montet, & Ansdell (2020), JOSS
• Feinstein et al. (2020, arXiv)

Bug reports and contributions

stella is MIT-licensed.

Source and issues on GitHub:

• Repo
• Issues