Problem Statement: Simulate the reaction in HW1 Part C under identical conditions but for a tubular reactor of the same volume discretized into 200 perfectly mixed sections to simulate slight back-mixing. To accelerate the numerical simulation, implement the code with Google JAX AI framework, using jax.jit, and jax.jacobian in conjunction with scipy.integrate.solve_ivp. Create plots of the concentration and temperature profiles through the tubular reaction over 2 seconds and qualitatively explain the results.
