Ok, T. (2024). A Dynamic Programming package for discrete MDPs in JAX (Version 0.3.0) [Computer software]. https://github.com/TolgaOk/jaxdp