Pignatelli, E. Navix: Accelerated gridworld navigation with JAX [Computer software]. https://github.com/epignatelli/navix