Pollux contains JAX implementations of data-driven models for stellar spectroscopy, like Lux and the Cannon.