pollux

pollux#

Introduction#

Pollux is a Python library for constructing generative models of astronomical spectra and other kinds of data. It is built on top of JAX and is designed for use in probabilistic and machine learning contexts.

Two classes of models are currently supported:

  • Lux: Multi-output, generative, latent variable models for inferring embedded representations of spectroscopic and many other kinds of data.

  • Cannon: Data-driven models for inferring stellar parameters, element abundances, and other labels from stellar spectra.

Installation#

Install pollux with pip directly from the GitHub repository:

pip install git+https://github.com/adrn/pollux

In the future, we plan to provide stable releases through PyPI.

Get Started#

The best way to get started with pollux is to work through the tutorials:

Tutorials