DiffiLQRAX: Differentiable iLQR in JAX - Mullen, Thomas Soares; Schimel, Marine (2025)