suppression de jax et optimisation du code