jaxoplanet is a functional-programming-forward implementation of many features from the exoplanet and starry packages built on top of JAX (ascl:2111.002). It includes fast and robust implementations of many exoplanet-specific operations, including solving Kepler’s equation, and computing limb-darkened light curves. Built on top of JAX, jaxoplanet has first-class support for hardware acceleration using GPUs and TPUs, and integrates seamlessly with modeling tools such as NumPyro and Flax (ascl:2504.026).
Please see citation information at https://github.com/exoplanet-dev/jaxoplanet