jax

13c965a7 - Merge pull request #4872 from bchetioui:test_cumsum_cumprod