A Schrodinger-Poisson+ Solver in JAX - View it on GitHub