بهینه‌سازی و تضمین کیفیت در محاسبات عددی با JAX