This reverts a previous change which broke parameter AD support. However, this change breaks compat with integrator AD (unless chunk sizes align).

Trade off is worth it since integrator AD is not generally that useful.